LCOV - code coverage report
Current view: top level - drivers/iommu - dma-iommu.c (source / functions) Hit Total Coverage
Test: combined.info Lines: 2 386 0.5 %
Date: 2022-03-28 15:32:58 Functions: 1 34 2.9 %
Branches: 0 220 0.0 %

           Branch data     Line data    Source code
       1                 :            : // SPDX-License-Identifier: GPL-2.0-only
       2                 :            : /*
       3                 :            :  * A fairly generic DMA-API to IOMMU-API glue layer.
       4                 :            :  *
       5                 :            :  * Copyright (C) 2014-2015 ARM Ltd.
       6                 :            :  *
       7                 :            :  * based in part on arch/arm/mm/dma-mapping.c:
       8                 :            :  * Copyright (C) 2000-2004 Russell King
       9                 :            :  */
      10                 :            : 
      11                 :            : #include <linux/acpi_iort.h>
      12                 :            : #include <linux/device.h>
      13                 :            : #include <linux/dma-contiguous.h>
      14                 :            : #include <linux/dma-iommu.h>
      15                 :            : #include <linux/dma-noncoherent.h>
      16                 :            : #include <linux/gfp.h>
      17                 :            : #include <linux/huge_mm.h>
      18                 :            : #include <linux/iommu.h>
      19                 :            : #include <linux/iova.h>
      20                 :            : #include <linux/irq.h>
      21                 :            : #include <linux/mm.h>
      22                 :            : #include <linux/mutex.h>
      23                 :            : #include <linux/pci.h>
      24                 :            : #include <linux/scatterlist.h>
      25                 :            : #include <linux/vmalloc.h>
      26                 :            : #include <linux/crash_dump.h>
      27                 :            : 
      28                 :            : struct iommu_dma_msi_page {
      29                 :            :         struct list_head        list;
      30                 :            :         dma_addr_t              iova;
      31                 :            :         phys_addr_t             phys;
      32                 :            : };
      33                 :            : 
      34                 :            : enum iommu_dma_cookie_type {
      35                 :            :         IOMMU_DMA_IOVA_COOKIE,
      36                 :            :         IOMMU_DMA_MSI_COOKIE,
      37                 :            : };
      38                 :            : 
      39                 :            : struct iommu_dma_cookie {
      40                 :            :         enum iommu_dma_cookie_type      type;
      41                 :            :         union {
      42                 :            :                 /* Full allocator for IOMMU_DMA_IOVA_COOKIE */
      43                 :            :                 struct iova_domain      iovad;
      44                 :            :                 /* Trivial linear page allocator for IOMMU_DMA_MSI_COOKIE */
      45                 :            :                 dma_addr_t              msi_iova;
      46                 :            :         };
      47                 :            :         struct list_head                msi_page_list;
      48                 :            : 
      49                 :            :         /* Domain for flush queue callback; NULL if flush queue not in use */
      50                 :            :         struct iommu_domain             *fq_domain;
      51                 :            : };
      52                 :            : 
      53                 :          0 : static inline size_t cookie_msi_granule(struct iommu_dma_cookie *cookie)
      54                 :            : {
      55                 :          0 :         if (cookie->type == IOMMU_DMA_IOVA_COOKIE)
      56                 :          0 :                 return cookie->iovad.granule;
      57                 :            :         return PAGE_SIZE;
      58                 :            : }
      59                 :            : 
      60                 :          0 : static struct iommu_dma_cookie *cookie_alloc(enum iommu_dma_cookie_type type)
      61                 :            : {
      62                 :          0 :         struct iommu_dma_cookie *cookie;
      63                 :            : 
      64                 :          0 :         cookie = kzalloc(sizeof(*cookie), GFP_KERNEL);
      65   [ #  #  #  # ]:          0 :         if (cookie) {
      66                 :          0 :                 INIT_LIST_HEAD(&cookie->msi_page_list);
      67                 :          0 :                 cookie->type = type;
      68                 :            :         }
      69                 :          0 :         return cookie;
      70                 :            : }
      71                 :            : 
      72                 :            : /**
      73                 :            :  * iommu_get_dma_cookie - Acquire DMA-API resources for a domain
      74                 :            :  * @domain: IOMMU domain to prepare for DMA-API usage
      75                 :            :  *
      76                 :            :  * IOMMU drivers should normally call this from their domain_alloc
      77                 :            :  * callback when domain->type == IOMMU_DOMAIN_DMA.
      78                 :            :  */
      79                 :          0 : int iommu_get_dma_cookie(struct iommu_domain *domain)
      80                 :            : {
      81         [ #  # ]:          0 :         if (domain->iova_cookie)
      82                 :            :                 return -EEXIST;
      83                 :            : 
      84                 :          0 :         domain->iova_cookie = cookie_alloc(IOMMU_DMA_IOVA_COOKIE);
      85         [ #  # ]:          0 :         if (!domain->iova_cookie)
      86                 :          0 :                 return -ENOMEM;
      87                 :            : 
      88                 :            :         return 0;
      89                 :            : }
      90                 :            : EXPORT_SYMBOL(iommu_get_dma_cookie);
      91                 :            : 
      92                 :            : /**
      93                 :            :  * iommu_get_msi_cookie - Acquire just MSI remapping resources
      94                 :            :  * @domain: IOMMU domain to prepare
      95                 :            :  * @base: Start address of IOVA region for MSI mappings
      96                 :            :  *
      97                 :            :  * Users who manage their own IOVA allocation and do not want DMA API support,
      98                 :            :  * but would still like to take advantage of automatic MSI remapping, can use
      99                 :            :  * this to initialise their own domain appropriately. Users should reserve a
     100                 :            :  * contiguous IOVA region, starting at @base, large enough to accommodate the
     101                 :            :  * number of PAGE_SIZE mappings necessary to cover every MSI doorbell address
     102                 :            :  * used by the devices attached to @domain.
     103                 :            :  */
     104                 :          0 : int iommu_get_msi_cookie(struct iommu_domain *domain, dma_addr_t base)
     105                 :            : {
     106                 :          0 :         struct iommu_dma_cookie *cookie;
     107                 :            : 
     108         [ #  # ]:          0 :         if (domain->type != IOMMU_DOMAIN_UNMANAGED)
     109                 :            :                 return -EINVAL;
     110                 :            : 
     111         [ #  # ]:          0 :         if (domain->iova_cookie)
     112                 :            :                 return -EEXIST;
     113                 :            : 
     114                 :          0 :         cookie = cookie_alloc(IOMMU_DMA_MSI_COOKIE);
     115         [ #  # ]:          0 :         if (!cookie)
     116                 :            :                 return -ENOMEM;
     117                 :            : 
     118                 :          0 :         cookie->msi_iova = base;
     119                 :          0 :         domain->iova_cookie = cookie;
     120                 :          0 :         return 0;
     121                 :            : }
     122                 :            : EXPORT_SYMBOL(iommu_get_msi_cookie);
     123                 :            : 
     124                 :            : /**
     125                 :            :  * iommu_put_dma_cookie - Release a domain's DMA mapping resources
     126                 :            :  * @domain: IOMMU domain previously prepared by iommu_get_dma_cookie() or
     127                 :            :  *          iommu_get_msi_cookie()
     128                 :            :  *
     129                 :            :  * IOMMU drivers should normally call this from their domain_free callback.
     130                 :            :  */
     131                 :          0 : void iommu_put_dma_cookie(struct iommu_domain *domain)
     132                 :            : {
     133                 :          0 :         struct iommu_dma_cookie *cookie = domain->iova_cookie;
     134                 :          0 :         struct iommu_dma_msi_page *msi, *tmp;
     135                 :            : 
     136         [ #  # ]:          0 :         if (!cookie)
     137                 :            :                 return;
     138                 :            : 
     139   [ #  #  #  # ]:          0 :         if (cookie->type == IOMMU_DMA_IOVA_COOKIE && cookie->iovad.granule)
     140                 :          0 :                 put_iova_domain(&cookie->iovad);
     141                 :            : 
     142         [ #  # ]:          0 :         list_for_each_entry_safe(msi, tmp, &cookie->msi_page_list, list) {
     143                 :          0 :                 list_del(&msi->list);
     144                 :          0 :                 kfree(msi);
     145                 :            :         }
     146                 :          0 :         kfree(cookie);
     147                 :          0 :         domain->iova_cookie = NULL;
     148                 :            : }
     149                 :            : EXPORT_SYMBOL(iommu_put_dma_cookie);
     150                 :            : 
     151                 :            : /**
     152                 :            :  * iommu_dma_get_resv_regions - Reserved region driver helper
     153                 :            :  * @dev: Device from iommu_get_resv_regions()
     154                 :            :  * @list: Reserved region list from iommu_get_resv_regions()
     155                 :            :  *
     156                 :            :  * IOMMU drivers can use this to implement their .get_resv_regions callback
     157                 :            :  * for general non-IOMMU-specific reservations. Currently, this covers GICv3
     158                 :            :  * ITS region reservation on ACPI based ARM platforms that may require HW MSI
     159                 :            :  * reservation.
     160                 :            :  */
     161                 :          0 : void iommu_dma_get_resv_regions(struct device *dev, struct list_head *list)
     162                 :            : {
     163                 :            : 
     164                 :          0 :         if (!is_of_node(dev_iommu_fwspec_get(dev)->iommu_fwnode))
     165                 :          0 :                 iort_iommu_msi_get_resv_regions(dev, list);
     166                 :            : 
     167                 :          0 : }
     168                 :            : EXPORT_SYMBOL(iommu_dma_get_resv_regions);
     169                 :            : 
     170                 :          0 : static int cookie_init_hw_msi_region(struct iommu_dma_cookie *cookie,
     171                 :            :                 phys_addr_t start, phys_addr_t end)
     172                 :            : {
     173                 :          0 :         struct iova_domain *iovad = &cookie->iovad;
     174                 :          0 :         struct iommu_dma_msi_page *msi_page;
     175                 :          0 :         int i, num_pages;
     176                 :            : 
     177                 :          0 :         start -= iova_offset(iovad, start);
     178                 :          0 :         num_pages = iova_align(iovad, end - start) >> iova_shift(iovad);
     179                 :            : 
     180         [ #  # ]:          0 :         for (i = 0; i < num_pages; i++) {
     181                 :          0 :                 msi_page = kmalloc(sizeof(*msi_page), GFP_KERNEL);
     182         [ #  # ]:          0 :                 if (!msi_page)
     183                 :            :                         return -ENOMEM;
     184                 :            : 
     185                 :          0 :                 msi_page->phys = start;
     186                 :          0 :                 msi_page->iova = start;
     187                 :          0 :                 INIT_LIST_HEAD(&msi_page->list);
     188                 :          0 :                 list_add(&msi_page->list, &cookie->msi_page_list);
     189                 :          0 :                 start += iovad->granule;
     190                 :            :         }
     191                 :            : 
     192                 :            :         return 0;
     193                 :            : }
     194                 :            : 
     195                 :          0 : static int iova_reserve_pci_windows(struct pci_dev *dev,
     196                 :            :                 struct iova_domain *iovad)
     197                 :            : {
     198                 :          0 :         struct pci_host_bridge *bridge = pci_find_host_bridge(dev->bus);
     199                 :          0 :         struct resource_entry *window;
     200                 :          0 :         unsigned long lo, hi;
     201                 :          0 :         phys_addr_t start = 0, end;
     202                 :            : 
     203         [ #  # ]:          0 :         resource_list_for_each_entry(window, &bridge->windows) {
     204         [ #  # ]:          0 :                 if (resource_type(window->res) != IORESOURCE_MEM)
     205                 :          0 :                         continue;
     206                 :            : 
     207                 :          0 :                 lo = iova_pfn(iovad, window->res->start - window->offset);
     208                 :          0 :                 hi = iova_pfn(iovad, window->res->end - window->offset);
     209                 :          0 :                 reserve_iova(iovad, lo, hi);
     210                 :            :         }
     211                 :            : 
     212                 :            :         /* Get reserved DMA windows from host bridge */
     213         [ #  # ]:          0 :         resource_list_for_each_entry(window, &bridge->dma_ranges) {
     214                 :          0 :                 end = window->res->start - window->offset;
     215                 :          0 : resv_iova:
     216         [ #  # ]:          0 :                 if (end > start) {
     217                 :          0 :                         lo = iova_pfn(iovad, start);
     218                 :          0 :                         hi = iova_pfn(iovad, end);
     219                 :          0 :                         reserve_iova(iovad, lo, hi);
     220                 :            :                 } else {
     221                 :            :                         /* dma_ranges list should be sorted */
     222                 :          0 :                         dev_err(&dev->dev, "Failed to reserve IOVA\n");
     223                 :          0 :                         return -EINVAL;
     224                 :            :                 }
     225                 :            : 
     226                 :          0 :                 start = window->res->end - window->offset + 1;
     227                 :            :                 /* If window is last entry */
     228   [ #  #  #  # ]:          0 :                 if (window->node.next == &bridge->dma_ranges &&
     229                 :            :                     end != ~(phys_addr_t)0) {
     230                 :          0 :                         end = ~(phys_addr_t)0;
     231                 :          0 :                         goto resv_iova;
     232                 :            :                 }
     233                 :            :         }
     234                 :            : 
     235                 :            :         return 0;
     236                 :            : }
     237                 :            : 
     238                 :            : static int iova_reserve_iommu_regions(struct device *dev,
     239                 :            :                 struct iommu_domain *domain)
     240                 :            : {
     241                 :            :         struct iommu_dma_cookie *cookie = domain->iova_cookie;
     242                 :            :         struct iova_domain *iovad = &cookie->iovad;
     243                 :            :         struct iommu_resv_region *region;
     244                 :            :         LIST_HEAD(resv_regions);
     245                 :            :         int ret = 0;
     246                 :            : 
     247                 :            :         if (dev_is_pci(dev)) {
     248                 :            :                 ret = iova_reserve_pci_windows(to_pci_dev(dev), iovad);
     249                 :            :                 if (ret)
     250                 :            :                         return ret;
     251                 :            :         }
     252                 :            : 
     253                 :            :         iommu_get_resv_regions(dev, &resv_regions);
     254                 :            :         list_for_each_entry(region, &resv_regions, list) {
     255                 :            :                 unsigned long lo, hi;
     256                 :            : 
     257                 :            :                 /* We ARE the software that manages these! */
     258                 :            :                 if (region->type == IOMMU_RESV_SW_MSI)
     259                 :            :                         continue;
     260                 :            : 
     261                 :            :                 lo = iova_pfn(iovad, region->start);
     262                 :            :                 hi = iova_pfn(iovad, region->start + region->length - 1);
     263                 :            :                 reserve_iova(iovad, lo, hi);
     264                 :            : 
     265                 :            :                 if (region->type == IOMMU_RESV_MSI)
     266                 :            :                         ret = cookie_init_hw_msi_region(cookie, region->start,
     267                 :            :                                         region->start + region->length);
     268                 :            :                 if (ret)
     269                 :            :                         break;
     270                 :            :         }
     271                 :            :         iommu_put_resv_regions(dev, &resv_regions);
     272                 :            : 
     273                 :            :         return ret;
     274                 :            : }
     275                 :            : 
     276                 :          0 : static void iommu_dma_flush_iotlb_all(struct iova_domain *iovad)
     277                 :            : {
     278                 :          0 :         struct iommu_dma_cookie *cookie;
     279                 :          0 :         struct iommu_domain *domain;
     280                 :            : 
     281                 :          0 :         cookie = container_of(iovad, struct iommu_dma_cookie, iovad);
     282                 :          0 :         domain = cookie->fq_domain;
     283                 :            :         /*
     284                 :            :          * The IOMMU driver supporting DOMAIN_ATTR_DMA_USE_FLUSH_QUEUE
     285                 :            :          * implies that ops->flush_iotlb_all must be non-NULL.
     286                 :            :          */
     287                 :          0 :         domain->ops->flush_iotlb_all(domain);
     288                 :          0 : }
     289                 :            : 
     290                 :            : /**
     291                 :            :  * iommu_dma_init_domain - Initialise a DMA mapping domain
     292                 :            :  * @domain: IOMMU domain previously prepared by iommu_get_dma_cookie()
     293                 :            :  * @base: IOVA at which the mappable address space starts
     294                 :            :  * @size: Size of IOVA space
     295                 :            :  * @dev: Device the domain is being initialised for
     296                 :            :  *
     297                 :            :  * @base and @size should be exact multiples of IOMMU page granularity to
     298                 :            :  * avoid rounding surprises. If necessary, we reserve the page at address 0
     299                 :            :  * to ensure it is an invalid IOVA. It is safe to reinitialise a domain, but
     300                 :            :  * any change which could make prior IOVAs invalid will fail.
     301                 :            :  */
     302                 :          0 : static int iommu_dma_init_domain(struct iommu_domain *domain, dma_addr_t base,
     303                 :            :                 u64 size, struct device *dev)
     304                 :            : {
     305                 :          0 :         struct iommu_dma_cookie *cookie = domain->iova_cookie;
     306                 :          0 :         unsigned long order, base_pfn;
     307                 :          0 :         struct iova_domain *iovad;
     308                 :          0 :         int attr;
     309                 :            : 
     310   [ #  #  #  # ]:          0 :         if (!cookie || cookie->type != IOMMU_DMA_IOVA_COOKIE)
     311                 :            :                 return -EINVAL;
     312                 :            : 
     313                 :          0 :         iovad = &cookie->iovad;
     314                 :            : 
     315                 :            :         /* Use the smallest supported page size for IOVA granularity */
     316         [ #  # ]:          0 :         order = __ffs(domain->pgsize_bitmap);
     317                 :          0 :         base_pfn = max_t(unsigned long, 1, base >> order);
     318                 :            : 
     319                 :            :         /* Check the domain allows at least some access to the device... */
     320         [ #  # ]:          0 :         if (domain->geometry.force_aperture) {
     321         [ #  # ]:          0 :                 if (base > domain->geometry.aperture_end ||
     322         [ #  # ]:          0 :                     base + size <= domain->geometry.aperture_start) {
     323                 :          0 :                         pr_warn("specified DMA range outside IOMMU capability\n");
     324                 :          0 :                         return -EFAULT;
     325                 :            :                 }
     326                 :            :                 /* ...then finally give it a kicking to make sure it fits */
     327                 :          0 :                 base_pfn = max_t(unsigned long, base_pfn,
     328                 :            :                                 domain->geometry.aperture_start >> order);
     329                 :            :         }
     330                 :            : 
     331                 :            :         /* start_pfn is always nonzero for an already-initialised domain */
     332         [ #  # ]:          0 :         if (iovad->start_pfn) {
     333   [ #  #  #  # ]:          0 :                 if (1UL << order != iovad->granule ||
     334                 :            :                     base_pfn != iovad->start_pfn) {
     335                 :          0 :                         pr_warn("Incompatible range for DMA domain\n");
     336                 :          0 :                         return -EFAULT;
     337                 :            :                 }
     338                 :            : 
     339                 :            :                 return 0;
     340                 :            :         }
     341                 :            : 
     342                 :          0 :         init_iova_domain(iovad, 1UL << order, base_pfn);
     343                 :            : 
     344   [ #  #  #  # ]:          0 :         if (!cookie->fq_domain && !iommu_domain_get_attr(domain,
     345         [ #  # ]:          0 :                         DOMAIN_ATTR_DMA_USE_FLUSH_QUEUE, &attr) && attr) {
     346                 :          0 :                 cookie->fq_domain = domain;
     347                 :          0 :                 init_iova_flush_queue(iovad, iommu_dma_flush_iotlb_all, NULL);
     348                 :            :         }
     349                 :            : 
     350         [ #  # ]:          0 :         if (!dev)
     351                 :            :                 return 0;
     352                 :            : 
     353                 :          0 :         return iova_reserve_iommu_regions(dev, domain);
     354                 :            : }
     355                 :            : 
     356                 :          0 : static int iommu_dma_deferred_attach(struct device *dev,
     357                 :            :                 struct iommu_domain *domain)
     358                 :            : {
     359                 :          0 :         const struct iommu_ops *ops = domain->ops;
     360                 :            : 
     361         [ #  # ]:          0 :         if (!is_kdump_kernel())
     362                 :            :                 return 0;
     363                 :            : 
     364   [ #  #  #  # ]:          0 :         if (unlikely(ops->is_attach_deferred &&
     365                 :            :                         ops->is_attach_deferred(domain, dev)))
     366                 :          0 :                 return iommu_attach_device(domain, dev);
     367                 :            : 
     368                 :            :         return 0;
     369                 :            : }
     370                 :            : 
     371                 :            : /**
     372                 :            :  * dma_info_to_prot - Translate DMA API directions and attributes to IOMMU API
     373                 :            :  *                    page flags.
     374                 :            :  * @dir: Direction of DMA transfer
     375                 :            :  * @coherent: Is the DMA master cache-coherent?
     376                 :            :  * @attrs: DMA attributes for the mapping
     377                 :            :  *
     378                 :            :  * Return: corresponding IOMMU API page protection flags
     379                 :            :  */
     380                 :          0 : static int dma_info_to_prot(enum dma_data_direction dir, bool coherent,
     381                 :            :                      unsigned long attrs)
     382                 :            : {
     383                 :          0 :         int prot = coherent ? IOMMU_CACHE : 0;
     384                 :            : 
     385   [ #  #  #  # ]:          0 :         if (attrs & DMA_ATTR_PRIVILEGED)
     386                 :          0 :                 prot |= IOMMU_PRIV;
     387                 :            : 
     388   [ #  #  #  #  :          0 :         switch (dir) {
          #  #  #  #  #  
                #  #  # ]
     389                 :          0 :         case DMA_BIDIRECTIONAL:
     390                 :          0 :                 return prot | IOMMU_READ | IOMMU_WRITE;
     391                 :          0 :         case DMA_TO_DEVICE:
     392                 :          0 :                 return prot | IOMMU_READ;
     393                 :          0 :         case DMA_FROM_DEVICE:
     394                 :          0 :                 return prot | IOMMU_WRITE;
     395                 :            :         default:
     396                 :            :                 return 0;
     397                 :            :         }
     398                 :            : }
     399                 :            : 
     400                 :            : static dma_addr_t iommu_dma_alloc_iova(struct iommu_domain *domain,
     401                 :            :                 size_t size, u64 dma_limit, struct device *dev)
     402                 :            : {
     403                 :            :         struct iommu_dma_cookie *cookie = domain->iova_cookie;
     404                 :            :         struct iova_domain *iovad = &cookie->iovad;
     405                 :            :         unsigned long shift, iova_len, iova = 0;
     406                 :            : 
     407                 :            :         if (cookie->type == IOMMU_DMA_MSI_COOKIE) {
     408                 :            :                 cookie->msi_iova += size;
     409                 :            :                 return cookie->msi_iova - size;
     410                 :            :         }
     411                 :            : 
     412                 :            :         shift = iova_shift(iovad);
     413                 :            :         iova_len = size >> shift;
     414                 :            :         /*
     415                 :            :          * Freeing non-power-of-two-sized allocations back into the IOVA caches
     416                 :            :          * will come back to bite us badly, so we have to waste a bit of space
     417                 :            :          * rounding up anything cacheable to make sure that can't happen. The
     418                 :            :          * order of the unadjusted size will still match upon freeing.
     419                 :            :          */
     420                 :            :         if (iova_len < (1 << (IOVA_RANGE_CACHE_MAX_SIZE - 1)))
     421                 :            :                 iova_len = roundup_pow_of_two(iova_len);
     422                 :            : 
     423                 :            :         dma_limit = min_not_zero(dma_limit, dev->bus_dma_limit);
     424                 :            : 
     425                 :            :         if (domain->geometry.force_aperture)
     426                 :            :                 dma_limit = min(dma_limit, (u64)domain->geometry.aperture_end);
     427                 :            : 
     428                 :            :         /* Try to get PCI devices a SAC address */
     429                 :            :         if (dma_limit > DMA_BIT_MASK(32) && dev_is_pci(dev))
     430                 :            :                 iova = alloc_iova_fast(iovad, iova_len,
     431                 :            :                                        DMA_BIT_MASK(32) >> shift, false);
     432                 :            : 
     433                 :            :         if (!iova)
     434                 :            :                 iova = alloc_iova_fast(iovad, iova_len, dma_limit >> shift,
     435                 :            :                                        true);
     436                 :            : 
     437                 :            :         return (dma_addr_t)iova << shift;
     438                 :            : }
     439                 :            : 
     440                 :          0 : static void iommu_dma_free_iova(struct iommu_dma_cookie *cookie,
     441                 :            :                 dma_addr_t iova, size_t size)
     442                 :            : {
     443                 :          0 :         struct iova_domain *iovad = &cookie->iovad;
     444                 :            : 
     445                 :            :         /* The MSI case is only ever cleaning up its most recent allocation */
     446         [ #  # ]:          0 :         if (cookie->type == IOMMU_DMA_MSI_COOKIE)
     447                 :          0 :                 cookie->msi_iova -= size;
     448         [ #  # ]:          0 :         else if (cookie->fq_domain)  /* non-strict mode */
     449                 :          0 :                 queue_iova(iovad, iova_pfn(iovad, iova),
     450                 :          0 :                                 size >> iova_shift(iovad), 0);
     451                 :            :         else
     452                 :          0 :                 free_iova_fast(iovad, iova_pfn(iovad, iova),
     453                 :          0 :                                 size >> iova_shift(iovad));
     454                 :          0 : }
     455                 :            : 
     456                 :          0 : static void __iommu_dma_unmap(struct device *dev, dma_addr_t dma_addr,
     457                 :            :                 size_t size)
     458                 :            : {
     459                 :          0 :         struct iommu_domain *domain = iommu_get_dma_domain(dev);
     460                 :          0 :         struct iommu_dma_cookie *cookie = domain->iova_cookie;
     461                 :          0 :         struct iova_domain *iovad = &cookie->iovad;
     462                 :          0 :         size_t iova_off = iova_offset(iovad, dma_addr);
     463                 :          0 :         struct iommu_iotlb_gather iotlb_gather;
     464                 :          0 :         size_t unmapped;
     465                 :            : 
     466                 :          0 :         dma_addr -= iova_off;
     467                 :          0 :         size = iova_align(iovad, size + iova_off);
     468                 :          0 :         iommu_iotlb_gather_init(&iotlb_gather);
     469                 :            : 
     470                 :          0 :         unmapped = iommu_unmap_fast(domain, dma_addr, size, &iotlb_gather);
     471         [ #  # ]:          0 :         WARN_ON(unmapped != size);
     472                 :            : 
     473         [ #  # ]:          0 :         if (!cookie->fq_domain)
     474         [ #  # ]:          0 :                 iommu_tlb_sync(domain, &iotlb_gather);
     475                 :          0 :         iommu_dma_free_iova(cookie, dma_addr, size);
     476                 :          0 : }
     477                 :            : 
     478                 :          0 : static dma_addr_t __iommu_dma_map(struct device *dev, phys_addr_t phys,
     479                 :            :                 size_t size, int prot, u64 dma_mask)
     480                 :            : {
     481                 :          0 :         struct iommu_domain *domain = iommu_get_dma_domain(dev);
     482                 :          0 :         struct iommu_dma_cookie *cookie = domain->iova_cookie;
     483                 :          0 :         struct iova_domain *iovad = &cookie->iovad;
     484                 :          0 :         size_t iova_off = iova_offset(iovad, phys);
     485                 :          0 :         dma_addr_t iova;
     486                 :            : 
     487         [ #  # ]:          0 :         if (unlikely(iommu_dma_deferred_attach(dev, domain)))
     488                 :            :                 return DMA_MAPPING_ERROR;
     489                 :            : 
     490                 :          0 :         size = iova_align(iovad, size + iova_off);
     491                 :            : 
     492                 :          0 :         iova = iommu_dma_alloc_iova(domain, size, dma_mask, dev);
     493         [ #  # ]:          0 :         if (!iova)
     494                 :            :                 return DMA_MAPPING_ERROR;
     495                 :            : 
     496         [ #  # ]:          0 :         if (iommu_map_atomic(domain, iova, phys - iova_off, size, prot)) {
     497                 :          0 :                 iommu_dma_free_iova(cookie, iova, size);
     498                 :          0 :                 return DMA_MAPPING_ERROR;
     499                 :            :         }
     500                 :          0 :         return iova + iova_off;
     501                 :            : }
     502                 :            : 
     503                 :            : static void __iommu_dma_free_pages(struct page **pages, int count)
     504                 :            : {
     505                 :            :         while (count--)
     506                 :            :                 __free_page(pages[count]);
     507                 :            :         kvfree(pages);
     508                 :            : }
     509                 :            : 
     510                 :            : static struct page **__iommu_dma_alloc_pages(struct device *dev,
     511                 :            :                 unsigned int count, unsigned long order_mask, gfp_t gfp)
     512                 :            : {
     513                 :            :         struct page **pages;
     514                 :            :         unsigned int i = 0, nid = dev_to_node(dev);
     515                 :            : 
     516                 :            :         order_mask &= (2U << MAX_ORDER) - 1;
     517                 :            :         if (!order_mask)
     518                 :            :                 return NULL;
     519                 :            : 
     520                 :            :         pages = kvzalloc(count * sizeof(*pages), GFP_KERNEL);
     521                 :            :         if (!pages)
     522                 :            :                 return NULL;
     523                 :            : 
     524                 :            :         /* IOMMU can map any pages, so himem can also be used here */
     525                 :            :         gfp |= __GFP_NOWARN | __GFP_HIGHMEM;
     526                 :            : 
     527                 :            :         while (count) {
     528                 :            :                 struct page *page = NULL;
     529                 :            :                 unsigned int order_size;
     530                 :            : 
     531                 :            :                 /*
     532                 :            :                  * Higher-order allocations are a convenience rather
     533                 :            :                  * than a necessity, hence using __GFP_NORETRY until
     534                 :            :                  * falling back to minimum-order allocations.
     535                 :            :                  */
     536                 :            :                 for (order_mask &= (2U << __fls(count)) - 1;
     537                 :            :                      order_mask; order_mask &= ~order_size) {
     538                 :            :                         unsigned int order = __fls(order_mask);
     539                 :            :                         gfp_t alloc_flags = gfp;
     540                 :            : 
     541                 :            :                         order_size = 1U << order;
     542                 :            :                         if (order_mask > order_size)
     543                 :            :                                 alloc_flags |= __GFP_NORETRY;
     544                 :            :                         page = alloc_pages_node(nid, alloc_flags, order);
     545                 :            :                         if (!page)
     546                 :            :                                 continue;
     547                 :            :                         if (!order)
     548                 :            :                                 break;
     549                 :            :                         if (!PageCompound(page)) {
     550                 :            :                                 split_page(page, order);
     551                 :            :                                 break;
     552                 :            :                         } else if (!split_huge_page(page)) {
     553                 :            :                                 break;
     554                 :            :                         }
     555                 :            :                         __free_pages(page, order);
     556                 :            :                 }
     557                 :            :                 if (!page) {
     558                 :            :                         __iommu_dma_free_pages(pages, i);
     559                 :            :                         return NULL;
     560                 :            :                 }
     561                 :            :                 count -= order_size;
     562                 :            :                 while (order_size--)
     563                 :            :                         pages[i++] = page++;
     564                 :            :         }
     565                 :            :         return pages;
     566                 :            : }
     567                 :            : 
     568                 :            : /**
     569                 :            :  * iommu_dma_alloc_remap - Allocate and map a buffer contiguous in IOVA space
     570                 :            :  * @dev: Device to allocate memory for. Must be a real device
     571                 :            :  *       attached to an iommu_dma_domain
     572                 :            :  * @size: Size of buffer in bytes
     573                 :            :  * @dma_handle: Out argument for allocated DMA handle
     574                 :            :  * @gfp: Allocation flags
     575                 :            :  * @attrs: DMA attributes for this allocation
     576                 :            :  *
     577                 :            :  * If @size is less than PAGE_SIZE, then a full CPU page will be allocated,
     578                 :            :  * but an IOMMU which supports smaller pages might not map the whole thing.
     579                 :            :  *
     580                 :            :  * Return: Mapped virtual address, or NULL on failure.
     581                 :            :  */
     582                 :            : static void *iommu_dma_alloc_remap(struct device *dev, size_t size,
     583                 :            :                 dma_addr_t *dma_handle, gfp_t gfp, unsigned long attrs)
     584                 :            : {
     585                 :            :         struct iommu_domain *domain = iommu_get_dma_domain(dev);
     586                 :            :         struct iommu_dma_cookie *cookie = domain->iova_cookie;
     587                 :            :         struct iova_domain *iovad = &cookie->iovad;
     588                 :            :         bool coherent = dev_is_dma_coherent(dev);
     589                 :            :         int ioprot = dma_info_to_prot(DMA_BIDIRECTIONAL, coherent, attrs);
     590                 :            :         pgprot_t prot = dma_pgprot(dev, PAGE_KERNEL, attrs);
     591                 :            :         unsigned int count, min_size, alloc_sizes = domain->pgsize_bitmap;
     592                 :            :         struct page **pages;
     593                 :            :         struct sg_table sgt;
     594                 :            :         dma_addr_t iova;
     595                 :            :         void *vaddr;
     596                 :            : 
     597                 :            :         *dma_handle = DMA_MAPPING_ERROR;
     598                 :            : 
     599                 :            :         if (unlikely(iommu_dma_deferred_attach(dev, domain)))
     600                 :            :                 return NULL;
     601                 :            : 
     602                 :            :         min_size = alloc_sizes & -alloc_sizes;
     603                 :            :         if (min_size < PAGE_SIZE) {
     604                 :            :                 min_size = PAGE_SIZE;
     605                 :            :                 alloc_sizes |= PAGE_SIZE;
     606                 :            :         } else {
     607                 :            :                 size = ALIGN(size, min_size);
     608                 :            :         }
     609                 :            :         if (attrs & DMA_ATTR_ALLOC_SINGLE_PAGES)
     610                 :            :                 alloc_sizes = min_size;
     611                 :            : 
     612                 :            :         count = PAGE_ALIGN(size) >> PAGE_SHIFT;
     613                 :            :         pages = __iommu_dma_alloc_pages(dev, count, alloc_sizes >> PAGE_SHIFT,
     614                 :            :                                         gfp);
     615                 :            :         if (!pages)
     616                 :            :                 return NULL;
     617                 :            : 
     618                 :            :         size = iova_align(iovad, size);
     619                 :            :         iova = iommu_dma_alloc_iova(domain, size, dev->coherent_dma_mask, dev);
     620                 :            :         if (!iova)
     621                 :            :                 goto out_free_pages;
     622                 :            : 
     623                 :            :         if (sg_alloc_table_from_pages(&sgt, pages, count, 0, size, GFP_KERNEL))
     624                 :            :                 goto out_free_iova;
     625                 :            : 
     626                 :            :         if (!(ioprot & IOMMU_CACHE)) {
     627                 :            :                 struct scatterlist *sg;
     628                 :            :                 int i;
     629                 :            : 
     630                 :            :                 for_each_sg(sgt.sgl, sg, sgt.orig_nents, i)
     631                 :            :                         arch_dma_prep_coherent(sg_page(sg), sg->length);
     632                 :            :         }
     633                 :            : 
     634                 :            :         if (iommu_map_sg_atomic(domain, iova, sgt.sgl, sgt.orig_nents, ioprot)
     635                 :            :                         < size)
     636                 :            :                 goto out_free_sg;
     637                 :            : 
     638                 :            :         vaddr = dma_common_pages_remap(pages, size, prot,
     639                 :            :                         __builtin_return_address(0));
     640                 :            :         if (!vaddr)
     641                 :            :                 goto out_unmap;
     642                 :            : 
     643                 :            :         *dma_handle = iova;
     644                 :            :         sg_free_table(&sgt);
     645                 :            :         return vaddr;
     646                 :            : 
     647                 :            : out_unmap:
     648                 :            :         __iommu_dma_unmap(dev, iova, size);
     649                 :            : out_free_sg:
     650                 :            :         sg_free_table(&sgt);
     651                 :            : out_free_iova:
     652                 :            :         iommu_dma_free_iova(cookie, iova, size);
     653                 :            : out_free_pages:
     654                 :            :         __iommu_dma_free_pages(pages, count);
     655                 :            :         return NULL;
     656                 :            : }
     657                 :            : 
     658                 :            : /**
     659                 :            :  * __iommu_dma_mmap - Map a buffer into provided user VMA
     660                 :            :  * @pages: Array representing buffer from __iommu_dma_alloc()
     661                 :            :  * @size: Size of buffer in bytes
     662                 :            :  * @vma: VMA describing requested userspace mapping
     663                 :            :  *
     664                 :            :  * Maps the pages of the buffer in @pages into @vma. The caller is responsible
     665                 :            :  * for verifying the correct size and protection of @vma beforehand.
     666                 :            :  */
     667                 :            : static int __iommu_dma_mmap(struct page **pages, size_t size,
     668                 :            :                 struct vm_area_struct *vma)
     669                 :            : {
     670                 :            :         return vm_map_pages(vma, pages, PAGE_ALIGN(size) >> PAGE_SHIFT);
     671                 :            : }
     672                 :            : 
     673                 :          0 : static void iommu_dma_sync_single_for_cpu(struct device *dev,
     674                 :            :                 dma_addr_t dma_handle, size_t size, enum dma_data_direction dir)
     675                 :            : {
     676                 :          0 :         phys_addr_t phys;
     677                 :            : 
     678                 :          0 :         if (dev_is_dma_coherent(dev))
     679                 :          0 :                 return;
     680                 :            : 
     681                 :            :         phys = iommu_iova_to_phys(iommu_get_dma_domain(dev), dma_handle);
     682                 :            :         arch_sync_dma_for_cpu(phys, size, dir);
     683                 :            : }
     684                 :            : 
     685                 :          0 : static void iommu_dma_sync_single_for_device(struct device *dev,
     686                 :            :                 dma_addr_t dma_handle, size_t size, enum dma_data_direction dir)
     687                 :            : {
     688                 :          0 :         phys_addr_t phys;
     689                 :            : 
     690                 :          0 :         if (dev_is_dma_coherent(dev))
     691                 :          0 :                 return;
     692                 :            : 
     693                 :            :         phys = iommu_iova_to_phys(iommu_get_dma_domain(dev), dma_handle);
     694                 :            :         arch_sync_dma_for_device(phys, size, dir);
     695                 :            : }
     696                 :            : 
     697                 :          0 : static void iommu_dma_sync_sg_for_cpu(struct device *dev,
     698                 :            :                 struct scatterlist *sgl, int nelems,
     699                 :            :                 enum dma_data_direction dir)
     700                 :            : {
     701                 :          0 :         struct scatterlist *sg;
     702                 :          0 :         int i;
     703                 :            : 
     704                 :          0 :         if (dev_is_dma_coherent(dev))
     705                 :          0 :                 return;
     706                 :            : 
     707                 :            :         for_each_sg(sgl, sg, nelems, i)
     708                 :            :                 arch_sync_dma_for_cpu(sg_phys(sg), sg->length, dir);
     709                 :            : }
     710                 :            : 
     711                 :          0 : static void iommu_dma_sync_sg_for_device(struct device *dev,
     712                 :            :                 struct scatterlist *sgl, int nelems,
     713                 :            :                 enum dma_data_direction dir)
     714                 :            : {
     715                 :          0 :         struct scatterlist *sg;
     716                 :          0 :         int i;
     717                 :            : 
     718                 :          0 :         if (dev_is_dma_coherent(dev))
     719                 :          0 :                 return;
     720                 :            : 
     721                 :            :         for_each_sg(sgl, sg, nelems, i)
     722                 :            :                 arch_sync_dma_for_device(sg_phys(sg), sg->length, dir);
     723                 :            : }
     724                 :            : 
     725                 :          0 : static dma_addr_t iommu_dma_map_page(struct device *dev, struct page *page,
     726                 :            :                 unsigned long offset, size_t size, enum dma_data_direction dir,
     727                 :            :                 unsigned long attrs)
     728                 :            : {
     729                 :          0 :         phys_addr_t phys = page_to_phys(page) + offset;
     730         [ #  # ]:          0 :         bool coherent = dev_is_dma_coherent(dev);
     731         [ #  # ]:          0 :         int prot = dma_info_to_prot(dir, coherent, attrs);
     732                 :          0 :         dma_addr_t dma_handle;
     733                 :            : 
     734         [ #  # ]:          0 :         dma_handle = __iommu_dma_map(dev, phys, size, prot, dma_get_mask(dev));
     735                 :          0 :         if (!coherent && !(attrs & DMA_ATTR_SKIP_CPU_SYNC) &&
     736                 :            :             dma_handle != DMA_MAPPING_ERROR)
     737                 :            :                 arch_sync_dma_for_device(phys, size, dir);
     738                 :          0 :         return dma_handle;
     739                 :            : }
     740                 :            : 
     741                 :          0 : static void iommu_dma_unmap_page(struct device *dev, dma_addr_t dma_handle,
     742                 :            :                 size_t size, enum dma_data_direction dir, unsigned long attrs)
     743                 :            : {
     744                 :          0 :         if (!(attrs & DMA_ATTR_SKIP_CPU_SYNC))
     745                 :            :                 iommu_dma_sync_single_for_cpu(dev, dma_handle, size, dir);
     746                 :          0 :         __iommu_dma_unmap(dev, dma_handle, size);
     747                 :          0 : }
     748                 :            : 
     749                 :            : /*
     750                 :            :  * Prepare a successfully-mapped scatterlist to give back to the caller.
     751                 :            :  *
     752                 :            :  * At this point the segments are already laid out by iommu_dma_map_sg() to
     753                 :            :  * avoid individually crossing any boundaries, so we merely need to check a
     754                 :            :  * segment's start address to avoid concatenating across one.
     755                 :            :  */
     756                 :          0 : static int __finalise_sg(struct device *dev, struct scatterlist *sg, int nents,
     757                 :            :                 dma_addr_t dma_addr)
     758                 :            : {
     759                 :          0 :         struct scatterlist *s, *cur = sg;
     760         [ #  # ]:          0 :         unsigned long seg_mask = dma_get_seg_boundary(dev);
     761         [ #  # ]:          0 :         unsigned int cur_len = 0, max_len = dma_get_max_seg_size(dev);
     762                 :          0 :         int i, count = 0;
     763                 :            : 
     764         [ #  # ]:          0 :         for_each_sg(sg, s, nents, i) {
     765                 :            :                 /* Restore this segment's original unaligned fields first */
     766                 :          0 :                 unsigned int s_iova_off = sg_dma_address(s);
     767                 :          0 :                 unsigned int s_length = sg_dma_len(s);
     768                 :          0 :                 unsigned int s_iova_len = s->length;
     769                 :            : 
     770                 :          0 :                 s->offset += s_iova_off;
     771                 :          0 :                 s->length = s_length;
     772                 :          0 :                 sg_dma_address(s) = DMA_MAPPING_ERROR;
     773                 :          0 :                 sg_dma_len(s) = 0;
     774                 :            : 
     775                 :            :                 /*
     776                 :            :                  * Now fill in the real DMA data. If...
     777                 :            :                  * - there is a valid output segment to append to
     778                 :            :                  * - and this segment starts on an IOVA page boundary
     779                 :            :                  * - but doesn't fall at a segment boundary
     780                 :            :                  * - and wouldn't make the resulting output segment too long
     781                 :            :                  */
     782   [ #  #  #  # ]:          0 :                 if (cur_len && !s_iova_off && (dma_addr & seg_mask) &&
     783         [ #  # ]:          0 :                     (max_len - cur_len >= s_length)) {
     784                 :            :                         /* ...then concatenate it with the previous one */
     785                 :          0 :                         cur_len += s_length;
     786                 :            :                 } else {
     787                 :            :                         /* Otherwise start the next output segment */
     788         [ #  # ]:          0 :                         if (i > 0)
     789                 :          0 :                                 cur = sg_next(cur);
     790                 :          0 :                         cur_len = s_length;
     791                 :          0 :                         count++;
     792                 :            : 
     793                 :          0 :                         sg_dma_address(cur) = dma_addr + s_iova_off;
     794                 :            :                 }
     795                 :            : 
     796                 :          0 :                 sg_dma_len(cur) = cur_len;
     797                 :          0 :                 dma_addr += s_iova_len;
     798                 :            : 
     799         [ #  # ]:          0 :                 if (s_length + s_iova_off < s_iova_len)
     800                 :          0 :                         cur_len = 0;
     801                 :            :         }
     802                 :          0 :         return count;
     803                 :            : }
     804                 :            : 
     805                 :            : /*
     806                 :            :  * If mapping failed, then just restore the original list,
     807                 :            :  * but making sure the DMA fields are invalidated.
     808                 :            :  */
     809                 :          0 : static void __invalidate_sg(struct scatterlist *sg, int nents)
     810                 :            : {
     811                 :          0 :         struct scatterlist *s;
     812                 :          0 :         int i;
     813                 :            : 
     814         [ #  # ]:          0 :         for_each_sg(sg, s, nents, i) {
     815         [ #  # ]:          0 :                 if (sg_dma_address(s) != DMA_MAPPING_ERROR)
     816                 :          0 :                         s->offset += sg_dma_address(s);
     817         [ #  # ]:          0 :                 if (sg_dma_len(s))
     818                 :          0 :                         s->length = sg_dma_len(s);
     819                 :          0 :                 sg_dma_address(s) = DMA_MAPPING_ERROR;
     820                 :          0 :                 sg_dma_len(s) = 0;
     821                 :            :         }
     822                 :          0 : }
     823                 :            : 
     824                 :            : /*
     825                 :            :  * The DMA API client is passing in a scatterlist which could describe
     826                 :            :  * any old buffer layout, but the IOMMU API requires everything to be
     827                 :            :  * aligned to IOMMU pages. Hence the need for this complicated bit of
     828                 :            :  * impedance-matching, to be able to hand off a suitably-aligned list,
     829                 :            :  * but still preserve the original offsets and sizes for the caller.
     830                 :            :  */
     831                 :          0 : static int iommu_dma_map_sg(struct device *dev, struct scatterlist *sg,
     832                 :            :                 int nents, enum dma_data_direction dir, unsigned long attrs)
     833                 :            : {
     834                 :          0 :         struct iommu_domain *domain = iommu_get_dma_domain(dev);
     835                 :          0 :         struct iommu_dma_cookie *cookie = domain->iova_cookie;
     836                 :          0 :         struct iova_domain *iovad = &cookie->iovad;
     837                 :          0 :         struct scatterlist *s, *prev = NULL;
     838         [ #  # ]:          0 :         int prot = dma_info_to_prot(dir, dev_is_dma_coherent(dev), attrs);
     839                 :          0 :         dma_addr_t iova;
     840                 :          0 :         size_t iova_len = 0;
     841         [ #  # ]:          0 :         unsigned long mask = dma_get_seg_boundary(dev);
     842                 :          0 :         int i;
     843                 :            : 
     844         [ #  # ]:          0 :         if (unlikely(iommu_dma_deferred_attach(dev, domain)))
     845                 :            :                 return 0;
     846                 :            : 
     847                 :            :         if (!(attrs & DMA_ATTR_SKIP_CPU_SYNC))
     848                 :            :                 iommu_dma_sync_sg_for_device(dev, sg, nents, dir);
     849                 :            : 
     850                 :            :         /*
     851                 :            :          * Work out how much IOVA space we need, and align the segments to
     852                 :            :          * IOVA granules for the IOMMU driver to handle. With some clever
     853                 :            :          * trickery we can modify the list in-place, but reversibly, by
     854                 :            :          * stashing the unaligned parts in the as-yet-unused DMA fields.
     855                 :            :          */
     856         [ #  # ]:          0 :         for_each_sg(sg, s, nents, i) {
     857         [ #  # ]:          0 :                 size_t s_iova_off = iova_offset(iovad, s->offset);
     858                 :          0 :                 size_t s_length = s->length;
     859                 :          0 :                 size_t pad_len = (mask - iova_len + 1) & mask;
     860                 :            : 
     861                 :          0 :                 sg_dma_address(s) = s_iova_off;
     862                 :          0 :                 sg_dma_len(s) = s_length;
     863                 :          0 :                 s->offset -= s_iova_off;
     864         [ #  # ]:          0 :                 s_length = iova_align(iovad, s_length + s_iova_off);
     865                 :          0 :                 s->length = s_length;
     866                 :            : 
     867                 :            :                 /*
     868                 :            :                  * Due to the alignment of our single IOVA allocation, we can
     869                 :            :                  * depend on these assumptions about the segment boundary mask:
     870                 :            :                  * - If mask size >= IOVA size, then the IOVA range cannot
     871                 :            :                  *   possibly fall across a boundary, so we don't care.
     872                 :            :                  * - If mask size < IOVA size, then the IOVA range must start
     873                 :            :                  *   exactly on a boundary, therefore we can lay things out
     874                 :            :                  *   based purely on segment lengths without needing to know
     875                 :            :                  *   the actual addresses beforehand.
     876                 :            :                  * - The mask must be a power of 2, so pad_len == 0 if
     877                 :            :                  *   iova_len == 0, thus we cannot dereference prev the first
     878                 :            :                  *   time through here (i.e. before it has a meaningful value).
     879                 :            :                  */
     880   [ #  #  #  # ]:          0 :                 if (pad_len && pad_len < s_length - 1) {
     881                 :          0 :                         prev->length += pad_len;
     882                 :          0 :                         iova_len += pad_len;
     883                 :            :                 }
     884                 :            : 
     885                 :          0 :                 iova_len += s_length;
     886                 :          0 :                 prev = s;
     887                 :            :         }
     888                 :            : 
     889         [ #  # ]:          0 :         iova = iommu_dma_alloc_iova(domain, iova_len, dma_get_mask(dev), dev);
     890         [ #  # ]:          0 :         if (!iova)
     891                 :          0 :                 goto out_restore_sg;
     892                 :            : 
     893                 :            :         /*
     894                 :            :          * We'll leave any physical concatenation to the IOMMU driver's
     895                 :            :          * implementation - it knows better than we do.
     896                 :            :          */
     897         [ #  # ]:          0 :         if (iommu_map_sg_atomic(domain, iova, sg, nents, prot) < iova_len)
     898                 :          0 :                 goto out_free_iova;
     899                 :            : 
     900                 :          0 :         return __finalise_sg(dev, sg, nents, iova);
     901                 :            : 
     902                 :            : out_free_iova:
     903                 :          0 :         iommu_dma_free_iova(cookie, iova, iova_len);
     904                 :          0 : out_restore_sg:
     905                 :          0 :         __invalidate_sg(sg, nents);
     906                 :          0 :         return 0;
     907                 :            : }
     908                 :            : 
     909                 :          0 : static void iommu_dma_unmap_sg(struct device *dev, struct scatterlist *sg,
     910                 :            :                 int nents, enum dma_data_direction dir, unsigned long attrs)
     911                 :            : {
     912                 :          0 :         dma_addr_t start, end;
     913                 :          0 :         struct scatterlist *tmp;
     914                 :          0 :         int i;
     915                 :            : 
     916                 :          0 :         if (!(attrs & DMA_ATTR_SKIP_CPU_SYNC))
     917                 :            :                 iommu_dma_sync_sg_for_cpu(dev, sg, nents, dir);
     918                 :            : 
     919                 :            :         /*
     920                 :            :          * The scatterlist segments are mapped into a single
     921                 :            :          * contiguous IOVA allocation, so this is incredibly easy.
     922                 :            :          */
     923                 :          0 :         start = sg_dma_address(sg);
     924         [ #  # ]:          0 :         for_each_sg(sg_next(sg), tmp, nents - 1, i) {
     925         [ #  # ]:          0 :                 if (sg_dma_len(tmp) == 0)
     926                 :            :                         break;
     927                 :          0 :                 sg = tmp;
     928                 :            :         }
     929                 :          0 :         end = sg_dma_address(sg) + sg_dma_len(sg);
     930                 :          0 :         __iommu_dma_unmap(dev, start, end - start);
     931                 :          0 : }
     932                 :            : 
     933                 :          0 : static dma_addr_t iommu_dma_map_resource(struct device *dev, phys_addr_t phys,
     934                 :            :                 size_t size, enum dma_data_direction dir, unsigned long attrs)
     935                 :            : {
     936         [ #  # ]:          0 :         return __iommu_dma_map(dev, phys, size,
     937                 :            :                         dma_info_to_prot(dir, false, attrs) | IOMMU_MMIO,
     938                 :            :                         dma_get_mask(dev));
     939                 :            : }
     940                 :            : 
     941                 :          0 : static void iommu_dma_unmap_resource(struct device *dev, dma_addr_t handle,
     942                 :            :                 size_t size, enum dma_data_direction dir, unsigned long attrs)
     943                 :            : {
     944                 :          0 :         __iommu_dma_unmap(dev, handle, size);
     945                 :          0 : }
     946                 :            : 
     947                 :            : static void __iommu_dma_free(struct device *dev, size_t size, void *cpu_addr)
     948                 :            : {
     949                 :            :         size_t alloc_size = PAGE_ALIGN(size);
     950                 :            :         int count = alloc_size >> PAGE_SHIFT;
     951                 :            :         struct page *page = NULL, **pages = NULL;
     952                 :            : 
     953                 :            :         /* Non-coherent atomic allocation? Easy */
     954                 :            :         if (IS_ENABLED(CONFIG_DMA_DIRECT_REMAP) &&
     955                 :            :             dma_free_from_pool(cpu_addr, alloc_size))
     956                 :            :                 return;
     957                 :            : 
     958                 :            :         if (IS_ENABLED(CONFIG_DMA_REMAP) && is_vmalloc_addr(cpu_addr)) {
     959                 :            :                 /*
     960                 :            :                  * If it the address is remapped, then it's either non-coherent
     961                 :            :                  * or highmem CMA, or an iommu_dma_alloc_remap() construction.
     962                 :            :                  */
     963                 :            :                 pages = dma_common_find_pages(cpu_addr);
     964                 :            :                 if (!pages)
     965                 :            :                         page = vmalloc_to_page(cpu_addr);
     966                 :            :                 dma_common_free_remap(cpu_addr, alloc_size);
     967                 :            :         } else {
     968                 :            :                 /* Lowmem means a coherent atomic or CMA allocation */
     969                 :            :                 page = virt_to_page(cpu_addr);
     970                 :            :         }
     971                 :            : 
     972                 :            :         if (pages)
     973                 :            :                 __iommu_dma_free_pages(pages, count);
     974                 :            :         if (page)
     975                 :            :                 dma_free_contiguous(dev, page, alloc_size);
     976                 :            : }
     977                 :            : 
     978                 :          0 : static void iommu_dma_free(struct device *dev, size_t size, void *cpu_addr,
     979                 :            :                 dma_addr_t handle, unsigned long attrs)
     980                 :            : {
     981                 :          0 :         __iommu_dma_unmap(dev, handle, size);
     982                 :          0 :         __iommu_dma_free(dev, size, cpu_addr);
     983                 :          0 : }
     984                 :            : 
     985                 :            : static void *iommu_dma_alloc_pages(struct device *dev, size_t size,
     986                 :            :                 struct page **pagep, gfp_t gfp, unsigned long attrs)
     987                 :            : {
     988                 :            :         bool coherent = dev_is_dma_coherent(dev);
     989                 :            :         size_t alloc_size = PAGE_ALIGN(size);
     990                 :            :         int node = dev_to_node(dev);
     991                 :            :         struct page *page = NULL;
     992                 :            :         void *cpu_addr;
     993                 :            : 
     994                 :            :         page = dma_alloc_contiguous(dev, alloc_size, gfp);
     995                 :            :         if (!page)
     996                 :            :                 page = alloc_pages_node(node, gfp, get_order(alloc_size));
     997                 :            :         if (!page)
     998                 :            :                 return NULL;
     999                 :            : 
    1000                 :            :         if (IS_ENABLED(CONFIG_DMA_REMAP) && (!coherent || PageHighMem(page))) {
    1001                 :            :                 pgprot_t prot = dma_pgprot(dev, PAGE_KERNEL, attrs);
    1002                 :            : 
    1003                 :            :                 cpu_addr = dma_common_contiguous_remap(page, alloc_size,
    1004                 :            :                                 prot, __builtin_return_address(0));
    1005                 :            :                 if (!cpu_addr)
    1006                 :            :                         goto out_free_pages;
    1007                 :            : 
    1008                 :            :                 if (!coherent)
    1009                 :            :                         arch_dma_prep_coherent(page, size);
    1010                 :            :         } else {
    1011                 :            :                 cpu_addr = page_address(page);
    1012                 :            :         }
    1013                 :            : 
    1014                 :            :         *pagep = page;
    1015                 :            :         memset(cpu_addr, 0, alloc_size);
    1016                 :            :         return cpu_addr;
    1017                 :            : out_free_pages:
    1018                 :            :         dma_free_contiguous(dev, page, alloc_size);
    1019                 :            :         return NULL;
    1020                 :            : }
    1021                 :            : 
    1022                 :          0 : static void *iommu_dma_alloc(struct device *dev, size_t size,
    1023                 :            :                 dma_addr_t *handle, gfp_t gfp, unsigned long attrs)
    1024                 :            : {
    1025         [ #  # ]:          0 :         bool coherent = dev_is_dma_coherent(dev);
    1026         [ #  # ]:          0 :         int ioprot = dma_info_to_prot(DMA_BIDIRECTIONAL, coherent, attrs);
    1027                 :          0 :         struct page *page = NULL;
    1028                 :          0 :         void *cpu_addr;
    1029                 :            : 
    1030                 :          0 :         gfp |= __GFP_ZERO;
    1031                 :            : 
    1032                 :          0 :         if (IS_ENABLED(CONFIG_DMA_REMAP) && gfpflags_allow_blocking(gfp) &&
    1033                 :            :             !(attrs & DMA_ATTR_FORCE_CONTIGUOUS))
    1034                 :            :                 return iommu_dma_alloc_remap(dev, size, handle, gfp, attrs);
    1035                 :            : 
    1036                 :          0 :         if (IS_ENABLED(CONFIG_DMA_DIRECT_REMAP) &&
    1037                 :            :             !gfpflags_allow_blocking(gfp) && !coherent)
    1038                 :            :                 cpu_addr = dma_alloc_from_pool(PAGE_ALIGN(size), &page, gfp);
    1039                 :            :         else
    1040                 :          0 :                 cpu_addr = iommu_dma_alloc_pages(dev, size, &page, gfp, attrs);
    1041         [ #  # ]:          0 :         if (!cpu_addr)
    1042                 :            :                 return NULL;
    1043                 :            : 
    1044                 :          0 :         *handle = __iommu_dma_map(dev, page_to_phys(page), size, ioprot,
    1045                 :            :                         dev->coherent_dma_mask);
    1046         [ #  # ]:          0 :         if (*handle == DMA_MAPPING_ERROR) {
    1047                 :          0 :                 __iommu_dma_free(dev, size, cpu_addr);
    1048                 :          0 :                 return NULL;
    1049                 :            :         }
    1050                 :            : 
    1051                 :            :         return cpu_addr;
    1052                 :            : }
    1053                 :            : 
    1054                 :          0 : static int iommu_dma_mmap(struct device *dev, struct vm_area_struct *vma,
    1055                 :            :                 void *cpu_addr, dma_addr_t dma_addr, size_t size,
    1056                 :            :                 unsigned long attrs)
    1057                 :            : {
    1058                 :          0 :         unsigned long nr_pages = PAGE_ALIGN(size) >> PAGE_SHIFT;
    1059                 :          0 :         unsigned long pfn, off = vma->vm_pgoff;
    1060                 :          0 :         int ret;
    1061                 :            : 
    1062                 :          0 :         vma->vm_page_prot = dma_pgprot(dev, vma->vm_page_prot, attrs);
    1063                 :            : 
    1064                 :          0 :         if (dma_mmap_from_dev_coherent(dev, vma, cpu_addr, size, &ret))
    1065                 :            :                 return ret;
    1066                 :            : 
    1067   [ #  #  #  # ]:          0 :         if (off >= nr_pages || vma_pages(vma) > nr_pages - off)
    1068                 :            :                 return -ENXIO;
    1069                 :            : 
    1070                 :          0 :         if (IS_ENABLED(CONFIG_DMA_REMAP) && is_vmalloc_addr(cpu_addr)) {
    1071                 :            :                 struct page **pages = dma_common_find_pages(cpu_addr);
    1072                 :            : 
    1073                 :            :                 if (pages)
    1074                 :            :                         return __iommu_dma_mmap(pages, size, vma);
    1075                 :            :                 pfn = vmalloc_to_pfn(cpu_addr);
    1076                 :            :         } else {
    1077         [ #  # ]:          0 :                 pfn = page_to_pfn(virt_to_page(cpu_addr));
    1078                 :            :         }
    1079                 :            : 
    1080                 :          0 :         return remap_pfn_range(vma, vma->vm_start, pfn + off,
    1081                 :            :                                vma->vm_end - vma->vm_start,
    1082                 :            :                                vma->vm_page_prot);
    1083                 :            : }
    1084                 :            : 
    1085                 :          0 : static int iommu_dma_get_sgtable(struct device *dev, struct sg_table *sgt,
    1086                 :            :                 void *cpu_addr, dma_addr_t dma_addr, size_t size,
    1087                 :            :                 unsigned long attrs)
    1088                 :            : {
    1089                 :          0 :         struct page *page;
    1090                 :          0 :         int ret;
    1091                 :            : 
    1092                 :          0 :         if (IS_ENABLED(CONFIG_DMA_REMAP) && is_vmalloc_addr(cpu_addr)) {
    1093                 :            :                 struct page **pages = dma_common_find_pages(cpu_addr);
    1094                 :            : 
    1095                 :            :                 if (pages) {
    1096                 :            :                         return sg_alloc_table_from_pages(sgt, pages,
    1097                 :            :                                         PAGE_ALIGN(size) >> PAGE_SHIFT,
    1098                 :            :                                         0, size, GFP_KERNEL);
    1099                 :            :                 }
    1100                 :            : 
    1101                 :            :                 page = vmalloc_to_page(cpu_addr);
    1102                 :            :         } else {
    1103         [ #  # ]:          0 :                 page = virt_to_page(cpu_addr);
    1104                 :            :         }
    1105                 :            : 
    1106                 :          0 :         ret = sg_alloc_table(sgt, 1, GFP_KERNEL);
    1107         [ #  # ]:          0 :         if (!ret)
    1108         [ #  # ]:          0 :                 sg_set_page(sgt->sgl, page, PAGE_ALIGN(size), 0);
    1109                 :          0 :         return ret;
    1110                 :            : }
    1111                 :            : 
    1112                 :          0 : static unsigned long iommu_dma_get_merge_boundary(struct device *dev)
    1113                 :            : {
    1114                 :          0 :         struct iommu_domain *domain = iommu_get_dma_domain(dev);
    1115                 :            : 
    1116                 :          0 :         return (1UL << __ffs(domain->pgsize_bitmap)) - 1;
    1117                 :            : }
    1118                 :            : 
    1119                 :            : static const struct dma_map_ops iommu_dma_ops = {
    1120                 :            :         .alloc                  = iommu_dma_alloc,
    1121                 :            :         .free                   = iommu_dma_free,
    1122                 :            :         .mmap                   = iommu_dma_mmap,
    1123                 :            :         .get_sgtable            = iommu_dma_get_sgtable,
    1124                 :            :         .map_page               = iommu_dma_map_page,
    1125                 :            :         .unmap_page             = iommu_dma_unmap_page,
    1126                 :            :         .map_sg                 = iommu_dma_map_sg,
    1127                 :            :         .unmap_sg               = iommu_dma_unmap_sg,
    1128                 :            :         .sync_single_for_cpu    = iommu_dma_sync_single_for_cpu,
    1129                 :            :         .sync_single_for_device = iommu_dma_sync_single_for_device,
    1130                 :            :         .sync_sg_for_cpu        = iommu_dma_sync_sg_for_cpu,
    1131                 :            :         .sync_sg_for_device     = iommu_dma_sync_sg_for_device,
    1132                 :            :         .map_resource           = iommu_dma_map_resource,
    1133                 :            :         .unmap_resource         = iommu_dma_unmap_resource,
    1134                 :            :         .get_merge_boundary     = iommu_dma_get_merge_boundary,
    1135                 :            : };
    1136                 :            : 
    1137                 :            : /*
    1138                 :            :  * The IOMMU core code allocates the default DMA domain, which the underlying
    1139                 :            :  * IOMMU driver needs to support via the dma-iommu layer.
    1140                 :            :  */
    1141                 :          0 : void iommu_setup_dma_ops(struct device *dev, u64 dma_base, u64 size)
    1142                 :            : {
    1143                 :          0 :         struct iommu_domain *domain = iommu_get_domain_for_dev(dev);
    1144                 :            : 
    1145         [ #  # ]:          0 :         if (!domain)
    1146                 :          0 :                 goto out_err;
    1147                 :            : 
    1148                 :            :         /*
    1149                 :            :          * The IOMMU core code allocates the default DMA domain, which the
    1150                 :            :          * underlying IOMMU driver needs to support via the dma-iommu layer.
    1151                 :            :          */
    1152         [ #  # ]:          0 :         if (domain->type == IOMMU_DOMAIN_DMA) {
    1153         [ #  # ]:          0 :                 if (iommu_dma_init_domain(domain, dma_base, size, dev))
    1154                 :          0 :                         goto out_err;
    1155                 :          0 :                 dev->dma_ops = &iommu_dma_ops;
    1156                 :            :         }
    1157                 :            : 
    1158                 :            :         return;
    1159                 :          0 : out_err:
    1160         [ #  # ]:          0 :          pr_warn("Failed to set up IOMMU for device %s; retaining platform DMA ops\n",
    1161                 :            :                  dev_name(dev));
    1162                 :            : }
    1163                 :            : 
    1164                 :          0 : static struct iommu_dma_msi_page *iommu_dma_get_msi_page(struct device *dev,
    1165                 :            :                 phys_addr_t msi_addr, struct iommu_domain *domain)
    1166                 :            : {
    1167                 :          0 :         struct iommu_dma_cookie *cookie = domain->iova_cookie;
    1168                 :          0 :         struct iommu_dma_msi_page *msi_page;
    1169                 :          0 :         dma_addr_t iova;
    1170                 :          0 :         int prot = IOMMU_WRITE | IOMMU_NOEXEC | IOMMU_MMIO;
    1171         [ #  # ]:          0 :         size_t size = cookie_msi_granule(cookie);
    1172                 :            : 
    1173                 :          0 :         msi_addr &= ~(phys_addr_t)(size - 1);
    1174         [ #  # ]:          0 :         list_for_each_entry(msi_page, &cookie->msi_page_list, list)
    1175         [ #  # ]:          0 :                 if (msi_page->phys == msi_addr)
    1176                 :          0 :                         return msi_page;
    1177                 :            : 
    1178                 :          0 :         msi_page = kzalloc(sizeof(*msi_page), GFP_KERNEL);
    1179         [ #  # ]:          0 :         if (!msi_page)
    1180                 :            :                 return NULL;
    1181                 :            : 
    1182         [ #  # ]:          0 :         iova = iommu_dma_alloc_iova(domain, size, dma_get_mask(dev), dev);
    1183         [ #  # ]:          0 :         if (!iova)
    1184                 :          0 :                 goto out_free_page;
    1185                 :            : 
    1186         [ #  # ]:          0 :         if (iommu_map(domain, iova, msi_addr, size, prot))
    1187                 :          0 :                 goto out_free_iova;
    1188                 :            : 
    1189                 :          0 :         INIT_LIST_HEAD(&msi_page->list);
    1190                 :          0 :         msi_page->phys = msi_addr;
    1191                 :          0 :         msi_page->iova = iova;
    1192                 :          0 :         list_add(&msi_page->list, &cookie->msi_page_list);
    1193                 :          0 :         return msi_page;
    1194                 :            : 
    1195                 :            : out_free_iova:
    1196                 :          0 :         iommu_dma_free_iova(cookie, iova, size);
    1197                 :          0 : out_free_page:
    1198                 :          0 :         kfree(msi_page);
    1199                 :          0 :         return NULL;
    1200                 :            : }
    1201                 :            : 
    1202                 :          0 : int iommu_dma_prepare_msi(struct msi_desc *desc, phys_addr_t msi_addr)
    1203                 :            : {
    1204                 :          0 :         struct device *dev = msi_desc_to_dev(desc);
    1205                 :          0 :         struct iommu_domain *domain = iommu_get_domain_for_dev(dev);
    1206                 :          0 :         struct iommu_dma_msi_page *msi_page;
    1207                 :          0 :         static DEFINE_MUTEX(msi_prepare_lock); /* see below */
    1208                 :            : 
    1209   [ #  #  #  # ]:          0 :         if (!domain || !domain->iova_cookie) {
    1210                 :          0 :                 desc->iommu_cookie = NULL;
    1211                 :          0 :                 return 0;
    1212                 :            :         }
    1213                 :            : 
    1214                 :            :         /*
    1215                 :            :          * In fact the whole prepare operation should already be serialised by
    1216                 :            :          * irq_domain_mutex further up the callchain, but that's pretty subtle
    1217                 :            :          * on its own, so consider this locking as failsafe documentation...
    1218                 :            :          */
    1219                 :          0 :         mutex_lock(&msi_prepare_lock);
    1220                 :          0 :         msi_page = iommu_dma_get_msi_page(dev, msi_addr, domain);
    1221                 :          0 :         mutex_unlock(&msi_prepare_lock);
    1222                 :            : 
    1223         [ #  # ]:          0 :         msi_desc_set_iommu_cookie(desc, msi_page);
    1224                 :            : 
    1225         [ #  # ]:          0 :         if (!msi_page)
    1226                 :          0 :                 return -ENOMEM;
    1227                 :            :         return 0;
    1228                 :            : }
    1229                 :            : 
    1230                 :          0 : void iommu_dma_compose_msi_msg(struct msi_desc *desc,
    1231                 :            :                                struct msi_msg *msg)
    1232                 :            : {
    1233                 :          0 :         struct device *dev = msi_desc_to_dev(desc);
    1234                 :          0 :         const struct iommu_domain *domain = iommu_get_domain_for_dev(dev);
    1235                 :          0 :         const struct iommu_dma_msi_page *msi_page;
    1236                 :            : 
    1237         [ #  # ]:          0 :         msi_page = msi_desc_get_iommu_cookie(desc);
    1238                 :            : 
    1239   [ #  #  #  #  :          0 :         if (!domain || !domain->iova_cookie || WARN_ON(!msi_page))
             #  #  #  # ]
    1240                 :            :                 return;
    1241                 :            : 
    1242                 :          0 :         msg->address_hi = upper_32_bits(msi_page->iova);
    1243         [ #  # ]:          0 :         msg->address_lo &= cookie_msi_granule(domain->iova_cookie) - 1;
    1244                 :          0 :         msg->address_lo += lower_32_bits(msi_page->iova);
    1245                 :            : }
    1246                 :            : 
    1247                 :         28 : static int iommu_dma_init(void)
    1248                 :            : {
    1249                 :         28 :         return iova_cache_get();
    1250                 :            : }
    1251                 :            : arch_initcall(iommu_dma_init);

Generated by: LCOV version 1.14