drivers/iommu/msm_iommu.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /* Copyright (c) 2010-2011, Code Aurora Forum. All rights reserved.
   3  *
   4  * Author: Stepan Moskovchenko <[email protected]>
   5  */
   6
   7 #define pr_fmt(fmt)     KBUILD_MODNAME ": " fmt
   8 #include <linux/kernel.h>
   9 #include <linux/init.h>
  10 #include <linux/platform_device.h>
  11 #include <linux/errno.h>
  12 #include <linux/io.h>
  13 #include <linux/io-pgtable.h>
  14 #include <linux/interrupt.h>
  15 #include <linux/list.h>
  16 #include <linux/spinlock.h>
  17 #include <linux/slab.h>
  18 #include <linux/iommu.h>
  19 #include <linux/clk.h>
  20 #include <linux/err.h>
  21 #include <linux/of_iommu.h>
  22
  23 #include <asm/cacheflush.h>
  24 #include <linux/sizes.h>
  25
  26 #include "msm_iommu_hw-8xxx.h"
  27 #include "msm_iommu.h"
  28
  29 #define MRC(reg, processor, op1, crn, crm, op2)                         \
  30 __asm__ __volatile__ (                                                  \
  31 "   mrc   "   #processor "," #op1 ", %0,"  #crn "," #crm "," #op2 "\n"  \
  32 : "=r" (reg))
  33
  34 /* bitmap of the page sizes currently supported */
  35 #define MSM_IOMMU_PGSIZES       (SZ_4K | SZ_64K | SZ_1M | SZ_16M)
  36
  37 DEFINE_SPINLOCK(msm_iommu_lock);
  38 static LIST_HEAD(qcom_iommu_devices);
  39 static struct iommu_ops msm_iommu_ops;
  40
  41 struct msm_priv {
  42         struct list_head list_attached;
  43         struct iommu_domain domain;
  44         struct io_pgtable_cfg   cfg;
  45         struct io_pgtable_ops   *iop;
  46         struct device           *dev;
  47         spinlock_t              pgtlock; /* pagetable lock */
  48 };
  49
  50 static struct msm_priv *to_msm_priv(struct iommu_domain *dom)
  51 {
  52         return container_of(dom, struct msm_priv, domain);
  53 }
  54
  55 static int __enable_clocks(struct msm_iommu_dev *iommu)
  56 {
  57         int ret;
  58
  59         ret = clk_enable(iommu->pclk);
  60         if (ret)
  61                 goto fail;
  62
  63         if (iommu->clk) {
  64                 ret = clk_enable(iommu->clk);
  65                 if (ret)
  66                         clk_disable(iommu->pclk);
  67         }
  68 fail:
  69         return ret;
  70 }
  71
  72 static void __disable_clocks(struct msm_iommu_dev *iommu)
  73 {
  74         if (iommu->clk)
  75                 clk_disable(iommu->clk);
  76         clk_disable(iommu->pclk);
  77 }
  78
  79 static void msm_iommu_reset(void __iomem *base, int ncb)
  80 {
  81         int ctx;
  82
  83         SET_RPUE(base, 0);
  84         SET_RPUEIE(base, 0);
  85         SET_ESRRESTORE(base, 0);
  86         SET_TBE(base, 0);
  87         SET_CR(base, 0);
  88         SET_SPDMBE(base, 0);
  89         SET_TESTBUSCR(base, 0);
  90         SET_TLBRSW(base, 0);
  91         SET_GLOBAL_TLBIALL(base, 0);
  92         SET_RPU_ACR(base, 0);
  93         SET_TLBLKCRWE(base, 1);
  94
  95         for (ctx = 0; ctx < ncb; ctx++) {
  96                 SET_BPRCOSH(base, ctx, 0);
  97                 SET_BPRCISH(base, ctx, 0);
  98                 SET_BPRCNSH(base, ctx, 0);
  99                 SET_BPSHCFG(base, ctx, 0);
 100                 SET_BPMTCFG(base, ctx, 0);
 101                 SET_ACTLR(base, ctx, 0);
 102                 SET_SCTLR(base, ctx, 0);
 103                 SET_FSRRESTORE(base, ctx, 0);
 104                 SET_TTBR0(base, ctx, 0);
 105                 SET_TTBR1(base, ctx, 0);
 106                 SET_TTBCR(base, ctx, 0);
 107                 SET_BFBCR(base, ctx, 0);
 108                 SET_PAR(base, ctx, 0);
 109                 SET_FAR(base, ctx, 0);
 110                 SET_CTX_TLBIALL(base, ctx, 0);
 111                 SET_TLBFLPTER(base, ctx, 0);
 112                 SET_TLBSLPTER(base, ctx, 0);
 113                 SET_TLBLKCR(base, ctx, 0);
 114                 SET_CONTEXTIDR(base, ctx, 0);
 115         }
 116 }
 117
 118 static void __flush_iotlb(void *cookie)
 119 {
 120         struct msm_priv *priv = cookie;
 121         struct msm_iommu_dev *iommu = NULL;
 122         struct msm_iommu_ctx_dev *master;
 123         int ret = 0;
 124
 125         list_for_each_entry(iommu, &priv->list_attached, dom_node) {
 126                 ret = __enable_clocks(iommu);
 127                 if (ret)
 128                         goto fail;
 129
 130                 list_for_each_entry(master, &iommu->ctx_list, list)
 131                         SET_CTX_TLBIALL(iommu->base, master->num, 0);
 132
 133                 __disable_clocks(iommu);
 134         }
 135 fail:
 136         return;
 137 }
 138
 139 static void __flush_iotlb_range(unsigned long iova, size_t size,
 140                                 size_t granule, bool leaf, void *cookie)
 141 {
 142         struct msm_priv *priv = cookie;
 143         struct msm_iommu_dev *iommu = NULL;
 144         struct msm_iommu_ctx_dev *master;
 145         int ret = 0;
 146         int temp_size;
 147
 148         list_for_each_entry(iommu, &priv->list_attached, dom_node) {
 149                 ret = __enable_clocks(iommu);
 150                 if (ret)
 151                         goto fail;
 152
 153                 list_for_each_entry(master, &iommu->ctx_list, list) {
 154                         temp_size = size;
 155                         do {
 156                                 iova &= TLBIVA_VA;
 157                                 iova |= GET_CONTEXTIDR_ASID(iommu->base,
 158                                                             master->num);
 159                                 SET_TLBIVA(iommu->base, master->num, iova);
 160                                 iova += granule;
 161                         } while (temp_size -= granule);
 162                 }
 163
 164                 __disable_clocks(iommu);
 165         }
 166
 167 fail:
 168         return;
 169 }
 170
 171 static void __flush_iotlb_walk(unsigned long iova, size_t size,
 172                                size_t granule, void *cookie)
 173 {
 174         __flush_iotlb_range(iova, size, granule, false, cookie);
 175 }
 176
 177 static void __flush_iotlb_leaf(unsigned long iova, size_t size,
 178                                size_t granule, void *cookie)
 179 {
 180         __flush_iotlb_range(iova, size, granule, true, cookie);
 181 }
 182
 183 static void __flush_iotlb_page(struct iommu_iotlb_gather *gather,
 184                                unsigned long iova, size_t granule, void *cookie)
 185 {
 186         __flush_iotlb_range(iova, granule, granule, true, cookie);
 187 }
 188
 189 static const struct iommu_flush_ops msm_iommu_flush_ops = {
 190         .tlb_flush_all = __flush_iotlb,
 191         .tlb_flush_walk = __flush_iotlb_walk,
 192         .tlb_flush_leaf = __flush_iotlb_leaf,
 193         .tlb_add_page = __flush_iotlb_page,
 194 };
 195
 196 static int msm_iommu_alloc_ctx(unsigned long *map, int start, int end)
 197 {
 198         int idx;
 199
 200         do {
 201                 idx = find_next_zero_bit(map, end, start);
 202                 if (idx == end)
 203                         return -ENOSPC;
 204         } while (test_and_set_bit(idx, map));
 205
 206         return idx;
 207 }
 208
 209 static void msm_iommu_free_ctx(unsigned long *map, int idx)
 210 {
 211         clear_bit(idx, map);
 212 }
 213
 214 static void config_mids(struct msm_iommu_dev *iommu,
 215                         struct msm_iommu_ctx_dev *master)
 216 {
 217         int mid, ctx, i;
 218
 219         for (i = 0; i < master->num_mids; i++) {
 220                 mid = master->mids[i];
 221                 ctx = master->num;
 222
 223                 SET_M2VCBR_N(iommu->base, mid, 0);
 224                 SET_CBACR_N(iommu->base, ctx, 0);
 225
 226                 /* Set VMID = 0 */
 227                 SET_VMID(iommu->base, mid, 0);
 228
 229                 /* Set the context number for that MID to this context */
 230                 SET_CBNDX(iommu->base, mid, ctx);
 231
 232                 /* Set MID associated with this context bank to 0*/
 233                 SET_CBVMID(iommu->base, ctx, 0);
 234
 235                 /* Set the ASID for TLB tagging for this context */
 236                 SET_CONTEXTIDR_ASID(iommu->base, ctx, ctx);
 237
 238                 /* Set security bit override to be Non-secure */
 239                 SET_NSCFG(iommu->base, mid, 3);
 240         }
 241 }
 242
 243 static void __reset_context(void __iomem *base, int ctx)
 244 {
 245         SET_BPRCOSH(base, ctx, 0);
 246         SET_BPRCISH(base, ctx, 0);
 247         SET_BPRCNSH(base, ctx, 0);
 248         SET_BPSHCFG(base, ctx, 0);
 249         SET_BPMTCFG(base, ctx, 0);
 250         SET_ACTLR(base, ctx, 0);
 251         SET_SCTLR(base, ctx, 0);
 252         SET_FSRRESTORE(base, ctx, 0);
 253         SET_TTBR0(base, ctx, 0);
 254         SET_TTBR1(base, ctx, 0);
 255         SET_TTBCR(base, ctx, 0);
 256         SET_BFBCR(base, ctx, 0);
 257         SET_PAR(base, ctx, 0);
 258         SET_FAR(base, ctx, 0);
 259         SET_CTX_TLBIALL(base, ctx, 0);
 260         SET_TLBFLPTER(base, ctx, 0);
 261         SET_TLBSLPTER(base, ctx, 0);
 262         SET_TLBLKCR(base, ctx, 0);
 263 }
 264
 265 static void __program_context(void __iomem *base, int ctx,
 266                               struct msm_priv *priv)
 267 {
 268         __reset_context(base, ctx);
 269
 270         /* Turn on TEX Remap */
 271         SET_TRE(base, ctx, 1);
 272         SET_AFE(base, ctx, 1);
 273
 274         /* Set up HTW mode */
 275         /* TLB miss configuration: perform HTW on miss */
 276         SET_TLBMCFG(base, ctx, 0x3);
 277
 278         /* V2P configuration: HTW for access */
 279         SET_V2PCFG(base, ctx, 0x3);
 280
 281         SET_TTBCR(base, ctx, priv->cfg.arm_v7s_cfg.tcr);
 282         SET_TTBR0(base, ctx, priv->cfg.arm_v7s_cfg.ttbr);
 283         SET_TTBR1(base, ctx, 0);
 284
 285         /* Set prrr and nmrr */
 286         SET_PRRR(base, ctx, priv->cfg.arm_v7s_cfg.prrr);
 287         SET_NMRR(base, ctx, priv->cfg.arm_v7s_cfg.nmrr);
 288
 289         /* Invalidate the TLB for this context */
 290         SET_CTX_TLBIALL(base, ctx, 0);
 291
 292         /* Set interrupt number to "secure" interrupt */
 293         SET_IRPTNDX(base, ctx, 0);
 294
 295         /* Enable context fault interrupt */
 296         SET_CFEIE(base, ctx, 1);
 297
 298         /* Stall access on a context fault and let the handler deal with it */
 299         SET_CFCFG(base, ctx, 1);
 300
 301         /* Redirect all cacheable requests to L2 slave port. */
 302         SET_RCISH(base, ctx, 1);
 303         SET_RCOSH(base, ctx, 1);
 304         SET_RCNSH(base, ctx, 1);
 305
 306         /* Turn on BFB prefetch */
 307         SET_BFBDFE(base, ctx, 1);
 308
 309         /* Enable the MMU */
 310         SET_M(base, ctx, 1);
 311 }
 312
 313 static struct iommu_domain *msm_iommu_domain_alloc(unsigned type)
 314 {
 315         struct msm_priv *priv;
 316
 317         if (type != IOMMU_DOMAIN_UNMANAGED)
 318                 return NULL;
 319
 320         priv = kzalloc(sizeof(*priv), GFP_KERNEL);
 321         if (!priv)
 322                 goto fail_nomem;
 323
 324         INIT_LIST_HEAD(&priv->list_attached);
 325
 326         priv->domain.geometry.aperture_start = 0;
 327         priv->domain.geometry.aperture_end   = (1ULL << 32) - 1;
 328         priv->domain.geometry.force_aperture = true;
 329
 330         return &priv->domain;
 331
 332 fail_nomem:
 333         kfree(priv);
 334         return NULL;
 335 }
 336
 337 static void msm_iommu_domain_free(struct iommu_domain *domain)
 338 {
 339         struct msm_priv *priv;
 340         unsigned long flags;
 341
 342         spin_lock_irqsave(&msm_iommu_lock, flags);
 343         priv = to_msm_priv(domain);
 344         kfree(priv);
 345         spin_unlock_irqrestore(&msm_iommu_lock, flags);
 346 }
 347
 348 static int msm_iommu_domain_config(struct msm_priv *priv)
 349 {
 350         spin_lock_init(&priv->pgtlock);
 351
 352         priv->cfg = (struct io_pgtable_cfg) {
 353                 .quirks = IO_PGTABLE_QUIRK_TLBI_ON_MAP,
 354                 .pgsize_bitmap = msm_iommu_ops.pgsize_bitmap,
 355                 .ias = 32,
 356                 .oas = 32,
 357                 .tlb = &msm_iommu_flush_ops,
 358                 .iommu_dev = priv->dev,
 359         };
 360
 361         priv->iop = alloc_io_pgtable_ops(ARM_V7S, &priv->cfg, priv);
 362         if (!priv->iop) {
 363                 dev_err(priv->dev, "Failed to allocate pgtable\n");
 364                 return -EINVAL;
 365         }
 366
 367         msm_iommu_ops.pgsize_bitmap = priv->cfg.pgsize_bitmap;
 368
 369         return 0;
 370 }
 371
 372 /* Must be called under msm_iommu_lock */
 373 static struct msm_iommu_dev *find_iommu_for_dev(struct device *dev)
 374 {
 375         struct msm_iommu_dev *iommu, *ret = NULL;
 376         struct msm_iommu_ctx_dev *master;
 377
 378         list_for_each_entry(iommu, &qcom_iommu_devices, dev_node) {
 379                 master = list_first_entry(&iommu->ctx_list,
 380                                           struct msm_iommu_ctx_dev,
 381                                           list);
 382                 if (master->of_node == dev->of_node) {
 383                         ret = iommu;
 384                         break;
 385                 }
 386         }
 387
 388         return ret;
 389 }
 390
 391 static int msm_iommu_add_device(struct device *dev)
 392 {
 393         struct msm_iommu_dev *iommu;
 394         struct iommu_group *group;
 395         unsigned long flags;
 396
 397         spin_lock_irqsave(&msm_iommu_lock, flags);
 398         iommu = find_iommu_for_dev(dev);
 399         spin_unlock_irqrestore(&msm_iommu_lock, flags);
 400
 401         if (iommu)
 402                 iommu_device_link(&iommu->iommu, dev);
 403         else
 404                 return -ENODEV;
 405
 406         group = iommu_group_get_for_dev(dev);
 407         if (IS_ERR(group))
 408                 return PTR_ERR(group);
 409
 410         iommu_group_put(group);
 411
 412         return 0;
 413 }
 414
 415 static void msm_iommu_remove_device(struct device *dev)
 416 {
 417         struct msm_iommu_dev *iommu;
 418         unsigned long flags;
 419
 420         spin_lock_irqsave(&msm_iommu_lock, flags);
 421         iommu = find_iommu_for_dev(dev);
 422         spin_unlock_irqrestore(&msm_iommu_lock, flags);
 423
 424         if (iommu)
 425                 iommu_device_unlink(&iommu->iommu, dev);
 426
 427         iommu_group_remove_device(dev);
 428 }
 429
 430 static int msm_iommu_attach_dev(struct iommu_domain *domain, struct device *dev)
 431 {
 432         int ret = 0;
 433         unsigned long flags;
 434         struct msm_iommu_dev *iommu;
 435         struct msm_priv *priv = to_msm_priv(domain);
 436         struct msm_iommu_ctx_dev *master;
 437
 438         priv->dev = dev;
 439         msm_iommu_domain_config(priv);
 440
 441         spin_lock_irqsave(&msm_iommu_lock, flags);
 442         list_for_each_entry(iommu, &qcom_iommu_devices, dev_node) {
 443                 master = list_first_entry(&iommu->ctx_list,
 444                                           struct msm_iommu_ctx_dev,
 445                                           list);
 446                 if (master->of_node == dev->of_node) {
 447                         ret = __enable_clocks(iommu);
 448                         if (ret)
 449                                 goto fail;
 450
 451                         list_for_each_entry(master, &iommu->ctx_list, list) {
 452                                 if (master->num) {
 453                                         dev_err(dev, "domain already attached");
 454                                         ret = -EEXIST;
 455                                         goto fail;
 456                                 }
 457                                 master->num =
 458                                         msm_iommu_alloc_ctx(iommu->context_map,
 459                                                             0, iommu->ncb);
 460                                 if (IS_ERR_VALUE(master->num)) {
 461                                         ret = -ENODEV;
 462                                         goto fail;
 463                                 }
 464                                 config_mids(iommu, master);
 465                                 __program_context(iommu->base, master->num,
 466                                                   priv);
 467                         }
 468                         __disable_clocks(iommu);
 469                         list_add(&iommu->dom_node, &priv->list_attached);
 470                 }
 471         }
 472
 473 fail:
 474         spin_unlock_irqrestore(&msm_iommu_lock, flags);
 475
 476         return ret;
 477 }
 478
 479 static void msm_iommu_detach_dev(struct iommu_domain *domain,
 480                                  struct device *dev)
 481 {
 482         struct msm_priv *priv = to_msm_priv(domain);
 483         unsigned long flags;
 484         struct msm_iommu_dev *iommu;
 485         struct msm_iommu_ctx_dev *master;
 486         int ret;
 487
 488         free_io_pgtable_ops(priv->iop);
 489
 490         spin_lock_irqsave(&msm_iommu_lock, flags);
 491         list_for_each_entry(iommu, &priv->list_attached, dom_node) {
 492                 ret = __enable_clocks(iommu);
 493                 if (ret)
 494                         goto fail;
 495
 496                 list_for_each_entry(master, &iommu->ctx_list, list) {
 497                         msm_iommu_free_ctx(iommu->context_map, master->num);
 498                         __reset_context(iommu->base, master->num);
 499                 }
 500                 __disable_clocks(iommu);
 501         }
 502 fail:
 503         spin_unlock_irqrestore(&msm_iommu_lock, flags);
 504 }
 505
 506 static int msm_iommu_map(struct iommu_domain *domain, unsigned long iova,
 507                          phys_addr_t pa, size_t len, int prot, gfp_t gfp)
 508 {
 509         struct msm_priv *priv = to_msm_priv(domain);
 510         unsigned long flags;
 511         int ret;
 512
 513         spin_lock_irqsave(&priv->pgtlock, flags);
 514         ret = priv->iop->map(priv->iop, iova, pa, len, prot);
 515         spin_unlock_irqrestore(&priv->pgtlock, flags);
 516
 517         return ret;
 518 }
 519
 520 static size_t msm_iommu_unmap(struct iommu_domain *domain, unsigned long iova,
 521                               size_t len, struct iommu_iotlb_gather *gather)
 522 {
 523         struct msm_priv *priv = to_msm_priv(domain);
 524         unsigned long flags;
 525
 526         spin_lock_irqsave(&priv->pgtlock, flags);
 527         len = priv->iop->unmap(priv->iop, iova, len, gather);
 528         spin_unlock_irqrestore(&priv->pgtlock, flags);
 529
 530         return len;
 531 }
 532
 533 static phys_addr_t msm_iommu_iova_to_phys(struct iommu_domain *domain,
 534                                           dma_addr_t va)
 535 {
 536         struct msm_priv *priv;
 537         struct msm_iommu_dev *iommu;
 538         struct msm_iommu_ctx_dev *master;
 539         unsigned int par;
 540         unsigned long flags;
 541         phys_addr_t ret = 0;
 542
 543         spin_lock_irqsave(&msm_iommu_lock, flags);
 544
 545         priv = to_msm_priv(domain);
 546         iommu = list_first_entry(&priv->list_attached,
 547                                  struct msm_iommu_dev, dom_node);
 548
 549         if (list_empty(&iommu->ctx_list))
 550                 goto fail;
 551
 552         master = list_first_entry(&iommu->ctx_list,
 553                                   struct msm_iommu_ctx_dev, list);
 554         if (!master)
 555                 goto fail;
 556
 557         ret = __enable_clocks(iommu);
 558         if (ret)
 559                 goto fail;
 560
 561         /* Invalidate context TLB */
 562         SET_CTX_TLBIALL(iommu->base, master->num, 0);
 563         SET_V2PPR(iommu->base, master->num, va & V2Pxx_VA);
 564
 565         par = GET_PAR(iommu->base, master->num);
 566
 567         /* We are dealing with a supersection */
 568         if (GET_NOFAULT_SS(iommu->base, master->num))
 569                 ret = (par & 0xFF000000) | (va & 0x00FFFFFF);
 570         else    /* Upper 20 bits from PAR, lower 12 from VA */
 571                 ret = (par & 0xFFFFF000) | (va & 0x00000FFF);
 572
 573         if (GET_FAULT(iommu->base, master->num))
 574                 ret = 0;
 575
 576         __disable_clocks(iommu);
 577 fail:
 578         spin_unlock_irqrestore(&msm_iommu_lock, flags);
 579         return ret;
 580 }
 581
 582 static bool msm_iommu_capable(enum iommu_cap cap)
 583 {
 584         return false;
 585 }
 586
 587 static void print_ctx_regs(void __iomem *base, int ctx)
 588 {
 589         unsigned int fsr = GET_FSR(base, ctx);
 590         pr_err("FAR    = %08x    PAR    = %08x\n",
 591                GET_FAR(base, ctx), GET_PAR(base, ctx));
 592         pr_err("FSR    = %08x [%s%s%s%s%s%s%s%s%s%s]\n", fsr,
 593                         (fsr & 0x02) ? "TF " : "",
 594                         (fsr & 0x04) ? "AFF " : "",
 595                         (fsr & 0x08) ? "APF " : "",
 596                         (fsr & 0x10) ? "TLBMF " : "",
 597                         (fsr & 0x20) ? "HTWDEEF " : "",
 598                         (fsr & 0x40) ? "HTWSEEF " : "",
 599                         (fsr & 0x80) ? "MHF " : "",
 600                         (fsr & 0x10000) ? "SL " : "",
 601                         (fsr & 0x40000000) ? "SS " : "",
 602                         (fsr & 0x80000000) ? "MULTI " : "");
 603
 604         pr_err("FSYNR0 = %08x    FSYNR1 = %08x\n",
 605                GET_FSYNR0(base, ctx), GET_FSYNR1(base, ctx));
 606         pr_err("TTBR0  = %08x    TTBR1  = %08x\n",
 607                GET_TTBR0(base, ctx), GET_TTBR1(base, ctx));
 608         pr_err("SCTLR  = %08x    ACTLR  = %08x\n",
 609                GET_SCTLR(base, ctx), GET_ACTLR(base, ctx));
 610 }
 611
 612 static void insert_iommu_master(struct device *dev,
 613                                 struct msm_iommu_dev **iommu,
 614                                 struct of_phandle_args *spec)
 615 {
 616         struct msm_iommu_ctx_dev *master = dev->archdata.iommu;
 617         int sid;
 618
 619         if (list_empty(&(*iommu)->ctx_list)) {
 620                 master = kzalloc(sizeof(*master), GFP_ATOMIC);
 621                 master->of_node = dev->of_node;
 622                 list_add(&master->list, &(*iommu)->ctx_list);
 623                 dev->archdata.iommu = master;
 624         }
 625
 626         for (sid = 0; sid < master->num_mids; sid++)
 627                 if (master->mids[sid] == spec->args[0]) {
 628                         dev_warn(dev, "Stream ID 0x%hx repeated; ignoring\n",
 629                                  sid);
 630                         return;
 631                 }
 632
 633         master->mids[master->num_mids++] = spec->args[0];
 634 }
 635
 636 static int qcom_iommu_of_xlate(struct device *dev,
 637                                struct of_phandle_args *spec)
 638 {
 639         struct msm_iommu_dev *iommu;
 640         unsigned long flags;
 641         int ret = 0;
 642
 643         spin_lock_irqsave(&msm_iommu_lock, flags);
 644         list_for_each_entry(iommu, &qcom_iommu_devices, dev_node)
 645                 if (iommu->dev->of_node == spec->np)
 646                         break;
 647
 648         if (!iommu || iommu->dev->of_node != spec->np) {
 649                 ret = -ENODEV;
 650                 goto fail;
 651         }
 652
 653         insert_iommu_master(dev, &iommu, spec);
 654 fail:
 655         spin_unlock_irqrestore(&msm_iommu_lock, flags);
 656
 657         return ret;
 658 }
 659
 660 irqreturn_t msm_iommu_fault_handler(int irq, void *dev_id)
 661 {
 662         struct msm_iommu_dev *iommu = dev_id;
 663         unsigned int fsr;
 664         int i, ret;
 665
 666         spin_lock(&msm_iommu_lock);
 667
 668         if (!iommu) {
 669                 pr_err("Invalid device ID in context interrupt handler\n");
 670                 goto fail;
 671         }
 672
 673         pr_err("Unexpected IOMMU page fault!\n");
 674         pr_err("base = %08x\n", (unsigned int)iommu->base);
 675
 676         ret = __enable_clocks(iommu);
 677         if (ret)
 678                 goto fail;
 679
 680         for (i = 0; i < iommu->ncb; i++) {
 681                 fsr = GET_FSR(iommu->base, i);
 682                 if (fsr) {
 683                         pr_err("Fault occurred in context %d.\n", i);
 684                         pr_err("Interesting registers:\n");
 685                         print_ctx_regs(iommu->base, i);
 686                         SET_FSR(iommu->base, i, 0x4000000F);
 687                 }
 688         }
 689         __disable_clocks(iommu);
 690 fail:
 691         spin_unlock(&msm_iommu_lock);
 692         return 0;
 693 }
 694
 695 static struct iommu_ops msm_iommu_ops = {
 696         .capable = msm_iommu_capable,
 697         .domain_alloc = msm_iommu_domain_alloc,
 698         .domain_free = msm_iommu_domain_free,
 699         .attach_dev = msm_iommu_attach_dev,
 700         .detach_dev = msm_iommu_detach_dev,
 701         .map = msm_iommu_map,
 702         .unmap = msm_iommu_unmap,
 703         /*
 704          * Nothing is needed here, the barrier to guarantee
 705          * completion of the tlb sync operation is implicitly
 706          * taken care when the iommu client does a writel before
 707          * kick starting the other master.
 708          */
 709         .iotlb_sync = NULL,
 710         .iova_to_phys = msm_iommu_iova_to_phys,
 711         .add_device = msm_iommu_add_device,
 712         .remove_device = msm_iommu_remove_device,
 713         .device_group = generic_device_group,
 714         .pgsize_bitmap = MSM_IOMMU_PGSIZES,
 715         .of_xlate = qcom_iommu_of_xlate,
 716 };
 717
 718 static int msm_iommu_probe(struct platform_device *pdev)
 719 {
 720         struct resource *r;
 721         resource_size_t ioaddr;
 722         struct msm_iommu_dev *iommu;
 723         int ret, par, val;
 724
 725         iommu = devm_kzalloc(&pdev->dev, sizeof(*iommu), GFP_KERNEL);
 726         if (!iommu)
 727                 return -ENODEV;
 728
 729         iommu->dev = &pdev->dev;
 730         INIT_LIST_HEAD(&iommu->ctx_list);
 731
 732         iommu->pclk = devm_clk_get(iommu->dev, "smmu_pclk");
 733         if (IS_ERR(iommu->pclk)) {
 734                 dev_err(iommu->dev, "could not get smmu_pclk\n");
 735                 return PTR_ERR(iommu->pclk);
 736         }
 737
 738         ret = clk_prepare(iommu->pclk);
 739         if (ret) {
 740                 dev_err(iommu->dev, "could not prepare smmu_pclk\n");
 741                 return ret;
 742         }
 743
 744         iommu->clk = devm_clk_get(iommu->dev, "iommu_clk");
 745         if (IS_ERR(iommu->clk)) {
 746                 dev_err(iommu->dev, "could not get iommu_clk\n");
 747                 clk_unprepare(iommu->pclk);
 748                 return PTR_ERR(iommu->clk);
 749         }
 750
 751         ret = clk_prepare(iommu->clk);
 752         if (ret) {
 753                 dev_err(iommu->dev, "could not prepare iommu_clk\n");
 754                 clk_unprepare(iommu->pclk);
 755                 return ret;
 756         }
 757
 758         r = platform_get_resource(pdev, IORESOURCE_MEM, 0);
 759         iommu->base = devm_ioremap_resource(iommu->dev, r);
 760         if (IS_ERR(iommu->base)) {
 761                 dev_err(iommu->dev, "could not get iommu base\n");
 762                 ret = PTR_ERR(iommu->base);
 763                 goto fail;
 764         }
 765         ioaddr = r->start;
 766
 767         iommu->irq = platform_get_irq(pdev, 0);
 768         if (iommu->irq < 0) {
 769                 ret = -ENODEV;
 770                 goto fail;
 771         }
 772
 773         ret = of_property_read_u32(iommu->dev->of_node, "qcom,ncb", &val);
 774         if (ret) {
 775                 dev_err(iommu->dev, "could not get ncb\n");
 776                 goto fail;
 777         }
 778         iommu->ncb = val;
 779
 780         msm_iommu_reset(iommu->base, iommu->ncb);
 781         SET_M(iommu->base, 0, 1);
 782         SET_PAR(iommu->base, 0, 0);
 783         SET_V2PCFG(iommu->base, 0, 1);
 784         SET_V2PPR(iommu->base, 0, 0);
 785         par = GET_PAR(iommu->base, 0);
 786         SET_V2PCFG(iommu->base, 0, 0);
 787         SET_M(iommu->base, 0, 0);
 788
 789         if (!par) {
 790                 pr_err("Invalid PAR value detected\n");
 791                 ret = -ENODEV;
 792                 goto fail;
 793         }
 794
 795         ret = devm_request_threaded_irq(iommu->dev, iommu->irq, NULL,
 796                                         msm_iommu_fault_handler,
 797                                         IRQF_ONESHOT | IRQF_SHARED,
 798                                         "msm_iommu_secure_irpt_handler",
 799                                         iommu);
 800         if (ret) {
 801                 pr_err("Request IRQ %d failed with ret=%d\n", iommu->irq, ret);
 802                 goto fail;
 803         }
 804
 805         list_add(&iommu->dev_node, &qcom_iommu_devices);
 806
 807         ret = iommu_device_sysfs_add(&iommu->iommu, iommu->dev, NULL,
 808                                      "msm-smmu.%pa", &ioaddr);
 809         if (ret) {
 810                 pr_err("Could not add msm-smmu at %pa to sysfs\n", &ioaddr);
 811                 goto fail;
 812         }
 813
 814         iommu_device_set_ops(&iommu->iommu, &msm_iommu_ops);
 815         iommu_device_set_fwnode(&iommu->iommu, &pdev->dev.of_node->fwnode);
 816
 817         ret = iommu_device_register(&iommu->iommu);
 818         if (ret) {
 819                 pr_err("Could not register msm-smmu at %pa\n", &ioaddr);
 820                 goto fail;
 821         }
 822
 823         bus_set_iommu(&platform_bus_type, &msm_iommu_ops);
 824
 825         pr_info("device mapped at %p, irq %d with %d ctx banks\n",
 826                 iommu->base, iommu->irq, iommu->ncb);
 827
 828         return ret;
 829 fail:
 830         clk_unprepare(iommu->clk);
 831         clk_unprepare(iommu->pclk);
 832         return ret;
 833 }
 834
 835 static const struct of_device_id msm_iommu_dt_match[] = {
 836         { .compatible = "qcom,apq8064-iommu" },
 837         {}
 838 };
 839
 840 static int msm_iommu_remove(struct platform_device *pdev)
 841 {
 842         struct msm_iommu_dev *iommu = platform_get_drvdata(pdev);
 843
 844         clk_unprepare(iommu->clk);
 845         clk_unprepare(iommu->pclk);
 846         return 0;
 847 }
 848
 849 static struct platform_driver msm_iommu_driver = {
 850         .driver = {
 851                 .name   = "msm_iommu",
 852                 .of_match_table = msm_iommu_dt_match,
 853         },
 854         .probe          = msm_iommu_probe,
 855         .remove         = msm_iommu_remove,
 856 };
 857
 858 static int __init msm_iommu_driver_init(void)
 859 {
 860         int ret;
 861
 862         ret = platform_driver_register(&msm_iommu_driver);
 863         if (ret != 0)
 864                 pr_err("Failed to register IOMMU driver\n");
 865
 866         return ret;
 867 }
 868 subsys_initcall(msm_iommu_driver_init);
 869