drivers/iommu/apple-dart.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * Apple DART (Device Address Resolution Table) IOMMU driver
   4  *
   5  * Copyright (C) 2021 The Asahi Linux Contributors
   6  *
   7  * Based on arm/arm-smmu/arm-ssmu.c and arm/arm-smmu-v3/arm-smmu-v3.c
   8  *  Copyright (C) 2013 ARM Limited
   9  *  Copyright (C) 2015 ARM Limited
  10  * and on exynos-iommu.c
  11  *  Copyright (c) 2011,2016 Samsung Electronics Co., Ltd.
  12  */
  13
  14 #include <linux/atomic.h>
  15 #include <linux/bitfield.h>
  16 #include <linux/clk.h>
  17 #include <linux/dev_printk.h>
  18 #include <linux/dma-iommu.h>
  19 #include <linux/dma-mapping.h>
  20 #include <linux/err.h>
  21 #include <linux/interrupt.h>
  22 #include <linux/io-pgtable.h>
  23 #include <linux/iommu.h>
  24 #include <linux/iopoll.h>
  25 #include <linux/module.h>
  26 #include <linux/of.h>
  27 #include <linux/of_address.h>
  28 #include <linux/of_iommu.h>
  29 #include <linux/of_platform.h>
  30 #include <linux/pci.h>
  31 #include <linux/platform_device.h>
  32 #include <linux/slab.h>
  33 #include <linux/swab.h>
  34 #include <linux/types.h>
  35
  36 #define DART_MAX_STREAMS 16
  37 #define DART_MAX_TTBR 4
  38 #define MAX_DARTS_PER_DEVICE 2
  39
  40 #define DART_STREAM_ALL 0xffff
  41
  42 #define DART_PARAMS1 0x00
  43 #define DART_PARAMS_PAGE_SHIFT GENMASK(27, 24)
  44
  45 #define DART_PARAMS2 0x04
  46 #define DART_PARAMS_BYPASS_SUPPORT BIT(0)
  47
  48 #define DART_STREAM_COMMAND 0x20
  49 #define DART_STREAM_COMMAND_BUSY BIT(2)
  50 #define DART_STREAM_COMMAND_INVALIDATE BIT(20)
  51
  52 #define DART_STREAM_SELECT 0x34
  53
  54 #define DART_ERROR 0x40
  55 #define DART_ERROR_STREAM GENMASK(27, 24)
  56 #define DART_ERROR_CODE GENMASK(11, 0)
  57 #define DART_ERROR_FLAG BIT(31)
  58
  59 #define DART_ERROR_READ_FAULT BIT(4)
  60 #define DART_ERROR_WRITE_FAULT BIT(3)
  61 #define DART_ERROR_NO_PTE BIT(2)
  62 #define DART_ERROR_NO_PMD BIT(1)
  63 #define DART_ERROR_NO_TTBR BIT(0)
  64
  65 #define DART_CONFIG 0x60
  66 #define DART_CONFIG_LOCK BIT(15)
  67
  68 #define DART_STREAM_COMMAND_BUSY_TIMEOUT 100
  69
  70 #define DART_ERROR_ADDR_HI 0x54
  71 #define DART_ERROR_ADDR_LO 0x50
  72
  73 #define DART_STREAMS_ENABLE 0xfc
  74
  75 #define DART_TCR(sid) (0x100 + 4 * (sid))
  76 #define DART_TCR_TRANSLATE_ENABLE BIT(7)
  77 #define DART_TCR_BYPASS0_ENABLE BIT(8)
  78 #define DART_TCR_BYPASS1_ENABLE BIT(12)
  79
  80 #define DART_TTBR(sid, idx) (0x200 + 16 * (sid) + 4 * (idx))
  81 #define DART_TTBR_VALID BIT(31)
  82 #define DART_TTBR_SHIFT 12
  83
  84 /*
  85  * Private structure associated with each DART device.
  86  *
  87  * @dev: device struct
  88  * @regs: mapped MMIO region
  89  * @irq: interrupt number, can be shared with other DARTs
  90  * @clks: clocks associated with this DART
  91  * @num_clks: number of @clks
  92  * @lock: lock for hardware operations involving this dart
  93  * @pgsize: pagesize supported by this DART
  94  * @supports_bypass: indicates if this DART supports bypass mode
  95  * @force_bypass: force bypass mode due to pagesize mismatch?
  96  * @sid2group: maps stream ids to iommu_groups
  97  * @iommu: iommu core device
  98  */
  99 struct apple_dart {
 100         struct device *dev;
 101
 102         void __iomem *regs;
 103
 104         int irq;
 105         struct clk_bulk_data *clks;
 106         int num_clks;
 107
 108         spinlock_t lock;
 109
 110         u32 pgsize;
 111         u32 supports_bypass : 1;
 112         u32 force_bypass : 1;
 113
 114         struct iommu_group *sid2group[DART_MAX_STREAMS];
 115         struct iommu_device iommu;
 116 };
 117
 118 /*
 119  * Convenience struct to identify streams.
 120  *
 121  * The normal variant is used inside apple_dart_master_cfg which isn't written
 122  * to concurrently.
 123  * The atomic variant is used inside apple_dart_domain where we have to guard
 124  * against races from potential parallel calls to attach/detach_device.
 125  * Note that even inside the atomic variant the apple_dart pointer is not
 126  * protected: This pointer is initialized once under the domain init mutex
 127  * and never changed again afterwards. Devices with different dart pointers
 128  * cannot be attached to the same domain.
 129  *
 130  * @dart dart pointer
 131  * @sid stream id bitmap
 132  */
 133 struct apple_dart_stream_map {
 134         struct apple_dart *dart;
 135         unsigned long sidmap;
 136 };
 137 struct apple_dart_atomic_stream_map {
 138         struct apple_dart *dart;
 139         atomic64_t sidmap;
 140 };
 141
 142 /*
 143  * This structure is attached to each iommu domain handled by a DART.
 144  *
 145  * @pgtbl_ops: pagetable ops allocated by io-pgtable
 146  * @finalized: true if the domain has been completely initialized
 147  * @init_lock: protects domain initialization
 148  * @stream_maps: streams attached to this domain (valid for DMA/UNMANAGED only)
 149  * @domain: core iommu domain pointer
 150  */
 151 struct apple_dart_domain {
 152         struct io_pgtable_ops *pgtbl_ops;
 153
 154         bool finalized;
 155         struct mutex init_lock;
 156         struct apple_dart_atomic_stream_map stream_maps[MAX_DARTS_PER_DEVICE];
 157
 158         struct iommu_domain domain;
 159 };
 160
 161 /*
 162  * This structure is attached to devices with dev_iommu_priv_set() on of_xlate
 163  * and contains a list of streams bound to this device.
 164  * So far the worst case seen is a single device with two streams
 165  * from different darts, such that this simple static array is enough.
 166  *
 167  * @streams: streams for this device
 168  */
 169 struct apple_dart_master_cfg {
 170         struct apple_dart_stream_map stream_maps[MAX_DARTS_PER_DEVICE];
 171 };
 172
 173 /*
 174  * Helper macro to iterate over apple_dart_master_cfg.stream_maps and
 175  * apple_dart_domain.stream_maps
 176  *
 177  * @i int used as loop variable
 178  * @base pointer to base struct (apple_dart_master_cfg or apple_dart_domain)
 179  * @stream pointer to the apple_dart_streams struct for each loop iteration
 180  */
 181 #define for_each_stream_map(i, base, stream_map)                               \
 182         for (i = 0, stream_map = &(base)->stream_maps[0];                      \
 183              i < MAX_DARTS_PER_DEVICE && stream_map->dart;                     \
 184              stream_map = &(base)->stream_maps[++i])
 185
 186 static struct platform_driver apple_dart_driver;
 187 static const struct iommu_ops apple_dart_iommu_ops;
 188
 189 static struct apple_dart_domain *to_dart_domain(struct iommu_domain *dom)
 190 {
 191         return container_of(dom, struct apple_dart_domain, domain);
 192 }
 193
 194 static void
 195 apple_dart_hw_enable_translation(struct apple_dart_stream_map *stream_map)
 196 {
 197         int sid;
 198
 199         for_each_set_bit(sid, &stream_map->sidmap, DART_MAX_STREAMS)
 200                 writel(DART_TCR_TRANSLATE_ENABLE,
 201                        stream_map->dart->regs + DART_TCR(sid));
 202 }
 203
 204 static void apple_dart_hw_disable_dma(struct apple_dart_stream_map *stream_map)
 205 {
 206         int sid;
 207
 208         for_each_set_bit(sid, &stream_map->sidmap, DART_MAX_STREAMS)
 209                 writel(0, stream_map->dart->regs + DART_TCR(sid));
 210 }
 211
 212 static void
 213 apple_dart_hw_enable_bypass(struct apple_dart_stream_map *stream_map)
 214 {
 215         int sid;
 216
 217         WARN_ON(!stream_map->dart->supports_bypass);
 218         for_each_set_bit(sid, &stream_map->sidmap, DART_MAX_STREAMS)
 219                 writel(DART_TCR_BYPASS0_ENABLE | DART_TCR_BYPASS1_ENABLE,
 220                        stream_map->dart->regs + DART_TCR(sid));
 221 }
 222
 223 static void apple_dart_hw_set_ttbr(struct apple_dart_stream_map *stream_map,
 224                                    u8 idx, phys_addr_t paddr)
 225 {
 226         int sid;
 227
 228         WARN_ON(paddr & ((1 << DART_TTBR_SHIFT) - 1));
 229         for_each_set_bit(sid, &stream_map->sidmap, DART_MAX_STREAMS)
 230                 writel(DART_TTBR_VALID | (paddr >> DART_TTBR_SHIFT),
 231                        stream_map->dart->regs + DART_TTBR(sid, idx));
 232 }
 233
 234 static void apple_dart_hw_clear_ttbr(struct apple_dart_stream_map *stream_map,
 235                                      u8 idx)
 236 {
 237         int sid;
 238
 239         for_each_set_bit(sid, &stream_map->sidmap, DART_MAX_STREAMS)
 240                 writel(0, stream_map->dart->regs + DART_TTBR(sid, idx));
 241 }
 242
 243 static void
 244 apple_dart_hw_clear_all_ttbrs(struct apple_dart_stream_map *stream_map)
 245 {
 246         int i;
 247
 248         for (i = 0; i < DART_MAX_TTBR; ++i)
 249                 apple_dart_hw_clear_ttbr(stream_map, i);
 250 }
 251
 252 static int
 253 apple_dart_hw_stream_command(struct apple_dart_stream_map *stream_map,
 254                              u32 command)
 255 {
 256         unsigned long flags;
 257         int ret;
 258         u32 command_reg;
 259
 260         spin_lock_irqsave(&stream_map->dart->lock, flags);
 261
 262         writel(stream_map->sidmap, stream_map->dart->regs + DART_STREAM_SELECT);
 263         writel(command, stream_map->dart->regs + DART_STREAM_COMMAND);
 264
 265         ret = readl_poll_timeout_atomic(
 266                 stream_map->dart->regs + DART_STREAM_COMMAND, command_reg,
 267                 !(command_reg & DART_STREAM_COMMAND_BUSY), 1,
 268                 DART_STREAM_COMMAND_BUSY_TIMEOUT);
 269
 270         spin_unlock_irqrestore(&stream_map->dart->lock, flags);
 271
 272         if (ret) {
 273                 dev_err(stream_map->dart->dev,
 274                         "busy bit did not clear after command %x for streams %lx\n",
 275                         command, stream_map->sidmap);
 276                 return ret;
 277         }
 278
 279         return 0;
 280 }
 281
 282 static int
 283 apple_dart_hw_invalidate_tlb(struct apple_dart_stream_map *stream_map)
 284 {
 285         return apple_dart_hw_stream_command(stream_map,
 286                                             DART_STREAM_COMMAND_INVALIDATE);
 287 }
 288
 289 static int apple_dart_hw_reset(struct apple_dart *dart)
 290 {
 291         u32 config;
 292         struct apple_dart_stream_map stream_map;
 293
 294         config = readl(dart->regs + DART_CONFIG);
 295         if (config & DART_CONFIG_LOCK) {
 296                 dev_err(dart->dev, "DART is locked down until reboot: %08x\n",
 297                         config);
 298                 return -EINVAL;
 299         }
 300
 301         stream_map.dart = dart;
 302         stream_map.sidmap = DART_STREAM_ALL;
 303         apple_dart_hw_disable_dma(&stream_map);
 304         apple_dart_hw_clear_all_ttbrs(&stream_map);
 305
 306         /* enable all streams globally since TCR is used to control isolation */
 307         writel(DART_STREAM_ALL, dart->regs + DART_STREAMS_ENABLE);
 308
 309         /* clear any pending errors before the interrupt is unmasked */
 310         writel(readl(dart->regs + DART_ERROR), dart->regs + DART_ERROR);
 311
 312         return apple_dart_hw_invalidate_tlb(&stream_map);
 313 }
 314
 315 static void apple_dart_domain_flush_tlb(struct apple_dart_domain *domain)
 316 {
 317         int i;
 318         struct apple_dart_atomic_stream_map *domain_stream_map;
 319         struct apple_dart_stream_map stream_map;
 320
 321         for_each_stream_map(i, domain, domain_stream_map) {
 322                 stream_map.dart = domain_stream_map->dart;
 323                 stream_map.sidmap = atomic64_read(&domain_stream_map->sidmap);
 324                 apple_dart_hw_invalidate_tlb(&stream_map);
 325         }
 326 }
 327
 328 static void apple_dart_flush_iotlb_all(struct iommu_domain *domain)
 329 {
 330         apple_dart_domain_flush_tlb(to_dart_domain(domain));
 331 }
 332
 333 static void apple_dart_iotlb_sync(struct iommu_domain *domain,
 334                                   struct iommu_iotlb_gather *gather)
 335 {
 336         apple_dart_domain_flush_tlb(to_dart_domain(domain));
 337 }
 338
 339 static void apple_dart_iotlb_sync_map(struct iommu_domain *domain,
 340                                       unsigned long iova, size_t size)
 341 {
 342         apple_dart_domain_flush_tlb(to_dart_domain(domain));
 343 }
 344
 345 static phys_addr_t apple_dart_iova_to_phys(struct iommu_domain *domain,
 346                                            dma_addr_t iova)
 347 {
 348         struct apple_dart_domain *dart_domain = to_dart_domain(domain);
 349         struct io_pgtable_ops *ops = dart_domain->pgtbl_ops;
 350
 351         if (!ops)
 352                 return 0;
 353
 354         return ops->iova_to_phys(ops, iova);
 355 }
 356
 357 static int apple_dart_map_pages(struct iommu_domain *domain, unsigned long iova,
 358                                 phys_addr_t paddr, size_t pgsize,
 359                                 size_t pgcount, int prot, gfp_t gfp,
 360                                 size_t *mapped)
 361 {
 362         struct apple_dart_domain *dart_domain = to_dart_domain(domain);
 363         struct io_pgtable_ops *ops = dart_domain->pgtbl_ops;
 364
 365         if (!ops)
 366                 return -ENODEV;
 367
 368         return ops->map_pages(ops, iova, paddr, pgsize, pgcount, prot, gfp,
 369                               mapped);
 370 }
 371
 372 static size_t apple_dart_unmap_pages(struct iommu_domain *domain,
 373                                      unsigned long iova, size_t pgsize,
 374                                      size_t pgcount,
 375                                      struct iommu_iotlb_gather *gather)
 376 {
 377         struct apple_dart_domain *dart_domain = to_dart_domain(domain);
 378         struct io_pgtable_ops *ops = dart_domain->pgtbl_ops;
 379
 380         return ops->unmap_pages(ops, iova, pgsize, pgcount, gather);
 381 }
 382
 383 static void
 384 apple_dart_setup_translation(struct apple_dart_domain *domain,
 385                              struct apple_dart_stream_map *stream_map)
 386 {
 387         int i;
 388         struct io_pgtable_cfg *pgtbl_cfg =
 389                 &io_pgtable_ops_to_pgtable(domain->pgtbl_ops)->cfg;
 390
 391         for (i = 0; i < pgtbl_cfg->apple_dart_cfg.n_ttbrs; ++i)
 392                 apple_dart_hw_set_ttbr(stream_map, i,
 393                                        pgtbl_cfg->apple_dart_cfg.ttbr[i]);
 394         for (; i < DART_MAX_TTBR; ++i)
 395                 apple_dart_hw_clear_ttbr(stream_map, i);
 396
 397         apple_dart_hw_enable_translation(stream_map);
 398         apple_dart_hw_invalidate_tlb(stream_map);
 399 }
 400
 401 static int apple_dart_finalize_domain(struct iommu_domain *domain,
 402                                       struct apple_dart_master_cfg *cfg)
 403 {
 404         struct apple_dart_domain *dart_domain = to_dart_domain(domain);
 405         struct apple_dart *dart = cfg->stream_maps[0].dart;
 406         struct io_pgtable_cfg pgtbl_cfg;
 407         int ret = 0;
 408         int i;
 409
 410         mutex_lock(&dart_domain->init_lock);
 411
 412         if (dart_domain->finalized)
 413                 goto done;
 414
 415         for (i = 0; i < MAX_DARTS_PER_DEVICE; ++i) {
 416                 dart_domain->stream_maps[i].dart = cfg->stream_maps[i].dart;
 417                 atomic64_set(&dart_domain->stream_maps[i].sidmap,
 418                              cfg->stream_maps[i].sidmap);
 419         }
 420
 421         pgtbl_cfg = (struct io_pgtable_cfg){
 422                 .pgsize_bitmap = dart->pgsize,
 423                 .ias = 32,
 424                 .oas = 36,
 425                 .coherent_walk = 1,
 426                 .iommu_dev = dart->dev,
 427         };
 428
 429         dart_domain->pgtbl_ops =
 430                 alloc_io_pgtable_ops(APPLE_DART, &pgtbl_cfg, domain);
 431         if (!dart_domain->pgtbl_ops) {
 432                 ret = -ENOMEM;
 433                 goto done;
 434         }
 435
 436         domain->pgsize_bitmap = pgtbl_cfg.pgsize_bitmap;
 437         domain->geometry.aperture_start = 0;
 438         domain->geometry.aperture_end = DMA_BIT_MASK(32);
 439         domain->geometry.force_aperture = true;
 440
 441         dart_domain->finalized = true;
 442
 443 done:
 444         mutex_unlock(&dart_domain->init_lock);
 445         return ret;
 446 }
 447
 448 static int
 449 apple_dart_mod_streams(struct apple_dart_atomic_stream_map *domain_maps,
 450                        struct apple_dart_stream_map *master_maps,
 451                        bool add_streams)
 452 {
 453         int i;
 454
 455         for (i = 0; i < MAX_DARTS_PER_DEVICE; ++i) {
 456                 if (domain_maps[i].dart != master_maps[i].dart)
 457                         return -EINVAL;
 458         }
 459
 460         for (i = 0; i < MAX_DARTS_PER_DEVICE; ++i) {
 461                 if (!domain_maps[i].dart)
 462                         break;
 463                 if (add_streams)
 464                         atomic64_or(master_maps[i].sidmap,
 465                                     &domain_maps[i].sidmap);
 466                 else
 467                         atomic64_and(~master_maps[i].sidmap,
 468                                      &domain_maps[i].sidmap);
 469         }
 470
 471         return 0;
 472 }
 473
 474 static int apple_dart_domain_add_streams(struct apple_dart_domain *domain,
 475                                          struct apple_dart_master_cfg *cfg)
 476 {
 477         return apple_dart_mod_streams(domain->stream_maps, cfg->stream_maps,
 478                                       true);
 479 }
 480
 481 static int apple_dart_domain_remove_streams(struct apple_dart_domain *domain,
 482                                             struct apple_dart_master_cfg *cfg)
 483 {
 484         return apple_dart_mod_streams(domain->stream_maps, cfg->stream_maps,
 485                                       false);
 486 }
 487
 488 static int apple_dart_attach_dev(struct iommu_domain *domain,
 489                                  struct device *dev)
 490 {
 491         int ret, i;
 492         struct apple_dart_stream_map *stream_map;
 493         struct apple_dart_master_cfg *cfg = dev_iommu_priv_get(dev);
 494         struct apple_dart_domain *dart_domain = to_dart_domain(domain);
 495
 496         if (cfg->stream_maps[0].dart->force_bypass &&
 497             domain->type != IOMMU_DOMAIN_IDENTITY)
 498                 return -EINVAL;
 499         if (!cfg->stream_maps[0].dart->supports_bypass &&
 500             domain->type == IOMMU_DOMAIN_IDENTITY)
 501                 return -EINVAL;
 502
 503         ret = apple_dart_finalize_domain(domain, cfg);
 504         if (ret)
 505                 return ret;
 506
 507         switch (domain->type) {
 508         case IOMMU_DOMAIN_DMA:
 509         case IOMMU_DOMAIN_UNMANAGED:
 510                 ret = apple_dart_domain_add_streams(dart_domain, cfg);
 511                 if (ret)
 512                         return ret;
 513
 514                 for_each_stream_map(i, cfg, stream_map)
 515                         apple_dart_setup_translation(dart_domain, stream_map);
 516                 break;
 517         case IOMMU_DOMAIN_BLOCKED:
 518                 for_each_stream_map(i, cfg, stream_map)
 519                         apple_dart_hw_disable_dma(stream_map);
 520                 break;
 521         case IOMMU_DOMAIN_IDENTITY:
 522                 for_each_stream_map(i, cfg, stream_map)
 523                         apple_dart_hw_enable_bypass(stream_map);
 524                 break;
 525         }
 526
 527         return ret;
 528 }
 529
 530 static void apple_dart_detach_dev(struct iommu_domain *domain,
 531                                   struct device *dev)
 532 {
 533         int i;
 534         struct apple_dart_stream_map *stream_map;
 535         struct apple_dart_master_cfg *cfg = dev_iommu_priv_get(dev);
 536         struct apple_dart_domain *dart_domain = to_dart_domain(domain);
 537
 538         for_each_stream_map(i, cfg, stream_map)
 539                 apple_dart_hw_disable_dma(stream_map);
 540
 541         if (domain->type == IOMMU_DOMAIN_DMA ||
 542             domain->type == IOMMU_DOMAIN_UNMANAGED)
 543                 apple_dart_domain_remove_streams(dart_domain, cfg);
 544 }
 545
 546 static struct iommu_device *apple_dart_probe_device(struct device *dev)
 547 {
 548         struct apple_dart_master_cfg *cfg = dev_iommu_priv_get(dev);
 549         struct apple_dart_stream_map *stream_map;
 550         int i;
 551
 552         if (!cfg)
 553                 return ERR_PTR(-ENODEV);
 554
 555         for_each_stream_map(i, cfg, stream_map)
 556                 device_link_add(
 557                         dev, stream_map->dart->dev,
 558                         DL_FLAG_PM_RUNTIME | DL_FLAG_AUTOREMOVE_SUPPLIER);
 559
 560         return &cfg->stream_maps[0].dart->iommu;
 561 }
 562
 563 static void apple_dart_release_device(struct device *dev)
 564 {
 565         struct apple_dart_master_cfg *cfg = dev_iommu_priv_get(dev);
 566
 567         if (!cfg)
 568                 return;
 569
 570         dev_iommu_priv_set(dev, NULL);
 571         kfree(cfg);
 572 }
 573
 574 static struct iommu_domain *apple_dart_domain_alloc(unsigned int type)
 575 {
 576         struct apple_dart_domain *dart_domain;
 577
 578         if (type != IOMMU_DOMAIN_DMA && type != IOMMU_DOMAIN_UNMANAGED &&
 579             type != IOMMU_DOMAIN_IDENTITY && type != IOMMU_DOMAIN_BLOCKED)
 580                 return NULL;
 581
 582         dart_domain = kzalloc(sizeof(*dart_domain), GFP_KERNEL);
 583         if (!dart_domain)
 584                 return NULL;
 585
 586         iommu_get_dma_cookie(&dart_domain->domain);
 587         mutex_init(&dart_domain->init_lock);
 588
 589         /* no need to allocate pgtbl_ops or do any other finalization steps */
 590         if (type == IOMMU_DOMAIN_IDENTITY || type == IOMMU_DOMAIN_BLOCKED)
 591                 dart_domain->finalized = true;
 592
 593         return &dart_domain->domain;
 594 }
 595
 596 static void apple_dart_domain_free(struct iommu_domain *domain)
 597 {
 598         struct apple_dart_domain *dart_domain = to_dart_domain(domain);
 599
 600         if (dart_domain->pgtbl_ops)
 601                 free_io_pgtable_ops(dart_domain->pgtbl_ops);
 602
 603         kfree(dart_domain);
 604 }
 605
 606 static int apple_dart_of_xlate(struct device *dev, struct of_phandle_args *args)
 607 {
 608         struct apple_dart_master_cfg *cfg = dev_iommu_priv_get(dev);
 609         struct platform_device *iommu_pdev = of_find_device_by_node(args->np);
 610         struct apple_dart *dart = platform_get_drvdata(iommu_pdev);
 611         struct apple_dart *cfg_dart;
 612         int i, sid;
 613
 614         if (args->args_count != 1)
 615                 return -EINVAL;
 616         sid = args->args[0];
 617
 618         if (!cfg)
 619                 cfg = kzalloc(sizeof(*cfg), GFP_KERNEL);
 620         if (!cfg)
 621                 return -ENOMEM;
 622         dev_iommu_priv_set(dev, cfg);
 623
 624         cfg_dart = cfg->stream_maps[0].dart;
 625         if (cfg_dart) {
 626                 if (cfg_dart->supports_bypass != dart->supports_bypass)
 627                         return -EINVAL;
 628                 if (cfg_dart->force_bypass != dart->force_bypass)
 629                         return -EINVAL;
 630                 if (cfg_dart->pgsize != dart->pgsize)
 631                         return -EINVAL;
 632         }
 633
 634         for (i = 0; i < MAX_DARTS_PER_DEVICE; ++i) {
 635                 if (cfg->stream_maps[i].dart == dart) {
 636                         cfg->stream_maps[i].sidmap |= 1 << sid;
 637                         return 0;
 638                 }
 639         }
 640         for (i = 0; i < MAX_DARTS_PER_DEVICE; ++i) {
 641                 if (!cfg->stream_maps[i].dart) {
 642                         cfg->stream_maps[i].dart = dart;
 643                         cfg->stream_maps[i].sidmap = 1 << sid;
 644                         return 0;
 645                 }
 646         }
 647
 648         return -EINVAL;
 649 }
 650
 651 static DEFINE_MUTEX(apple_dart_groups_lock);
 652
 653 static void apple_dart_release_group(void *iommu_data)
 654 {
 655         int i, sid;
 656         struct apple_dart_stream_map *stream_map;
 657         struct apple_dart_master_cfg *group_master_cfg = iommu_data;
 658
 659         mutex_lock(&apple_dart_groups_lock);
 660
 661         for_each_stream_map(i, group_master_cfg, stream_map)
 662                 for_each_set_bit(sid, &stream_map->sidmap, DART_MAX_STREAMS)
 663                         stream_map->dart->sid2group[sid] = NULL;
 664
 665         kfree(iommu_data);
 666         mutex_unlock(&apple_dart_groups_lock);
 667 }
 668
 669 static struct iommu_group *apple_dart_device_group(struct device *dev)
 670 {
 671         int i, sid;
 672         struct apple_dart_master_cfg *cfg = dev_iommu_priv_get(dev);
 673         struct apple_dart_stream_map *stream_map;
 674         struct apple_dart_master_cfg *group_master_cfg;
 675         struct iommu_group *group = NULL;
 676         struct iommu_group *res = ERR_PTR(-EINVAL);
 677
 678         mutex_lock(&apple_dart_groups_lock);
 679
 680         for_each_stream_map(i, cfg, stream_map) {
 681                 for_each_set_bit(sid, &stream_map->sidmap, DART_MAX_STREAMS) {
 682                         struct iommu_group *stream_group =
 683                                 stream_map->dart->sid2group[sid];
 684
 685                         if (group && group != stream_group) {
 686                                 res = ERR_PTR(-EINVAL);
 687                                 goto out;
 688                         }
 689
 690                         group = stream_group;
 691                 }
 692         }
 693
 694         if (group) {
 695                 res = iommu_group_ref_get(group);
 696                 goto out;
 697         }
 698
 699 #ifdef CONFIG_PCI
 700         if (dev_is_pci(dev))
 701                 group = pci_device_group(dev);
 702         else
 703 #endif
 704                 group = generic_device_group(dev);
 705
 706         res = ERR_PTR(-ENOMEM);
 707         if (!group)
 708                 goto out;
 709
 710         group_master_cfg = kzalloc(sizeof(*group_master_cfg), GFP_KERNEL);
 711         if (!group_master_cfg) {
 712                 iommu_group_put(group);
 713                 goto out;
 714         }
 715
 716         memcpy(group_master_cfg, cfg, sizeof(*group_master_cfg));
 717         iommu_group_set_iommudata(group, group_master_cfg,
 718                 apple_dart_release_group);
 719
 720         for_each_stream_map(i, cfg, stream_map)
 721                 for_each_set_bit(sid, &stream_map->sidmap, DART_MAX_STREAMS)
 722                         stream_map->dart->sid2group[sid] = group;
 723
 724         res = group;
 725
 726 out:
 727         mutex_unlock(&apple_dart_groups_lock);
 728         return res;
 729 }
 730
 731 static int apple_dart_def_domain_type(struct device *dev)
 732 {
 733         struct apple_dart_master_cfg *cfg = dev_iommu_priv_get(dev);
 734
 735         if (cfg->stream_maps[0].dart->force_bypass)
 736                 return IOMMU_DOMAIN_IDENTITY;
 737         if (!cfg->stream_maps[0].dart->supports_bypass)
 738                 return IOMMU_DOMAIN_DMA;
 739
 740         return 0;
 741 }
 742
 743 static const struct iommu_ops apple_dart_iommu_ops = {
 744         .domain_alloc = apple_dart_domain_alloc,
 745         .domain_free = apple_dart_domain_free,
 746         .attach_dev = apple_dart_attach_dev,
 747         .detach_dev = apple_dart_detach_dev,
 748         .map_pages = apple_dart_map_pages,
 749         .unmap_pages = apple_dart_unmap_pages,
 750         .flush_iotlb_all = apple_dart_flush_iotlb_all,
 751         .iotlb_sync = apple_dart_iotlb_sync,
 752         .iotlb_sync_map = apple_dart_iotlb_sync_map,
 753         .iova_to_phys = apple_dart_iova_to_phys,
 754         .probe_device = apple_dart_probe_device,
 755         .release_device = apple_dart_release_device,
 756         .device_group = apple_dart_device_group,
 757         .of_xlate = apple_dart_of_xlate,
 758         .def_domain_type = apple_dart_def_domain_type,
 759         .pgsize_bitmap = -1UL, /* Restricted during dart probe */
 760         .owner = THIS_MODULE,
 761 };
 762
 763 static irqreturn_t apple_dart_irq(int irq, void *dev)
 764 {
 765         struct apple_dart *dart = dev;
 766         const char *fault_name = NULL;
 767         u32 error = readl(dart->regs + DART_ERROR);
 768         u32 error_code = FIELD_GET(DART_ERROR_CODE, error);
 769         u32 addr_lo = readl(dart->regs + DART_ERROR_ADDR_LO);
 770         u32 addr_hi = readl(dart->regs + DART_ERROR_ADDR_HI);
 771         u64 addr = addr_lo | (((u64)addr_hi) << 32);
 772         u8 stream_idx = FIELD_GET(DART_ERROR_STREAM, error);
 773
 774         if (!(error & DART_ERROR_FLAG))
 775                 return IRQ_NONE;
 776
 777         /* there should only be a single bit set but let's use == to be sure */
 778         if (error_code == DART_ERROR_READ_FAULT)
 779                 fault_name = "READ FAULT";
 780         else if (error_code == DART_ERROR_WRITE_FAULT)
 781                 fault_name = "WRITE FAULT";
 782         else if (error_code == DART_ERROR_NO_PTE)
 783                 fault_name = "NO PTE FOR IOVA";
 784         else if (error_code == DART_ERROR_NO_PMD)
 785                 fault_name = "NO PMD FOR IOVA";
 786         else if (error_code == DART_ERROR_NO_TTBR)
 787                 fault_name = "NO TTBR FOR IOVA";
 788         else
 789                 fault_name = "unknown";
 790
 791         dev_err_ratelimited(
 792                 dart->dev,
 793                 "translation fault: status:0x%x stream:%d code:0x%x (%s) at 0x%llx",
 794                 error, stream_idx, error_code, fault_name, addr);
 795
 796         writel(error, dart->regs + DART_ERROR);
 797         return IRQ_HANDLED;
 798 }
 799
 800 static int apple_dart_set_bus_ops(const struct iommu_ops *ops)
 801 {
 802         int ret;
 803
 804         if (!iommu_present(&platform_bus_type)) {
 805                 ret = bus_set_iommu(&platform_bus_type, ops);
 806                 if (ret)
 807                         return ret;
 808         }
 809 #ifdef CONFIG_PCI
 810         if (!iommu_present(&pci_bus_type)) {
 811                 ret = bus_set_iommu(&pci_bus_type, ops);
 812                 if (ret) {
 813                         bus_set_iommu(&platform_bus_type, NULL);
 814                         return ret;
 815                 }
 816         }
 817 #endif
 818         return 0;
 819 }
 820
 821 static int apple_dart_probe(struct platform_device *pdev)
 822 {
 823         int ret;
 824         u32 dart_params[2];
 825         struct resource *res;
 826         struct apple_dart *dart;
 827         struct device *dev = &pdev->dev;
 828
 829         dart = devm_kzalloc(dev, sizeof(*dart), GFP_KERNEL);
 830         if (!dart)
 831                 return -ENOMEM;
 832
 833         dart->dev = dev;
 834         spin_lock_init(&dart->lock);
 835
 836         dart->regs = devm_platform_get_and_ioremap_resource(pdev, 0, &res);
 837         if (IS_ERR(dart->regs))
 838                 return PTR_ERR(dart->regs);
 839
 840         if (resource_size(res) < 0x4000) {
 841                 dev_err(dev, "MMIO region too small (%pr)\n", res);
 842                 return -EINVAL;
 843         }
 844
 845         dart->irq = platform_get_irq(pdev, 0);
 846         if (dart->irq < 0)
 847                 return -ENODEV;
 848
 849         ret = devm_clk_bulk_get_all(dev, &dart->clks);
 850         if (ret < 0)
 851                 return ret;
 852         dart->num_clks = ret;
 853
 854         ret = clk_bulk_prepare_enable(dart->num_clks, dart->clks);
 855         if (ret)
 856                 return ret;
 857
 858         ret = apple_dart_hw_reset(dart);
 859         if (ret)
 860                 goto err_clk_disable;
 861
 862         dart_params[0] = readl(dart->regs + DART_PARAMS1);
 863         dart_params[1] = readl(dart->regs + DART_PARAMS2);
 864         dart->pgsize = 1 << FIELD_GET(DART_PARAMS_PAGE_SHIFT, dart_params[0]);
 865         dart->supports_bypass = dart_params[1] & DART_PARAMS_BYPASS_SUPPORT;
 866         dart->force_bypass = dart->pgsize > PAGE_SIZE;
 867
 868         ret = request_irq(dart->irq, apple_dart_irq, IRQF_SHARED,
 869                           "apple-dart fault handler", dart);
 870         if (ret)
 871                 goto err_clk_disable;
 872
 873         platform_set_drvdata(pdev, dart);
 874
 875         ret = apple_dart_set_bus_ops(&apple_dart_iommu_ops);
 876         if (ret)
 877                 goto err_free_irq;
 878
 879         ret = iommu_device_sysfs_add(&dart->iommu, dev, NULL, "apple-dart.%s",
 880                                      dev_name(&pdev->dev));
 881         if (ret)
 882                 goto err_remove_bus_ops;
 883
 884         ret = iommu_device_register(&dart->iommu, &apple_dart_iommu_ops, dev);
 885         if (ret)
 886                 goto err_sysfs_remove;
 887
 888         dev_info(
 889                 &pdev->dev,
 890                 "DART [pagesize %x, bypass support: %d, bypass forced: %d] initialized\n",
 891                 dart->pgsize, dart->supports_bypass, dart->force_bypass);
 892         return 0;
 893
 894 err_sysfs_remove:
 895         iommu_device_sysfs_remove(&dart->iommu);
 896 err_remove_bus_ops:
 897         apple_dart_set_bus_ops(NULL);
 898 err_free_irq:
 899         free_irq(dart->irq, dart);
 900 err_clk_disable:
 901         clk_bulk_disable_unprepare(dart->num_clks, dart->clks);
 902
 903         return ret;
 904 }
 905
 906 static int apple_dart_remove(struct platform_device *pdev)
 907 {
 908         struct apple_dart *dart = platform_get_drvdata(pdev);
 909
 910         apple_dart_hw_reset(dart);
 911         free_irq(dart->irq, dart);
 912         apple_dart_set_bus_ops(NULL);
 913
 914         iommu_device_unregister(&dart->iommu);
 915         iommu_device_sysfs_remove(&dart->iommu);
 916
 917         clk_bulk_disable_unprepare(dart->num_clks, dart->clks);
 918
 919         return 0;
 920 }
 921
 922 static const struct of_device_id apple_dart_of_match[] = {
 923         { .compatible = "apple,t8103-dart", .data = NULL },
 924         {},
 925 };
 926 MODULE_DEVICE_TABLE(of, apple_dart_of_match);
 927
 928 static struct platform_driver apple_dart_driver = {
 929         .driver = {
 930                 .name                   = "apple-dart",
 931                 .of_match_table         = apple_dart_of_match,
 932                 .suppress_bind_attrs    = true,
 933         },
 934         .probe  = apple_dart_probe,
 935         .remove = apple_dart_remove,
 936 };
 937
 938 module_platform_driver(apple_dart_driver);
 939
 940 MODULE_DESCRIPTION("IOMMU API for Apple's DART");
 941 MODULE_AUTHOR("Sven Peter <sven@svenpeter.dev>");
 942 MODULE_LICENSE("GPL v2");