ARM: dma-mapping: IOVA API chagne
[linux-3.10.git] / arch / arm / mm / dma-mapping.c
1 /*
2  *  linux/arch/arm/mm/dma-mapping.c
3  *
4  *  Copyright (C) 2000-2004 Russell King
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License version 2 as
8  * published by the Free Software Foundation.
9  *
10  *  DMA uncached mapping support.
11  */
12 #include <linux/module.h>
13 #include <linux/mm.h>
14 #include <linux/gfp.h>
15 #include <linux/errno.h>
16 #include <linux/list.h>
17 #include <linux/init.h>
18 #include <linux/device.h>
19 #include <linux/dma-mapping.h>
20 #include <linux/dma-contiguous.h>
21 #include <linux/highmem.h>
22 #include <linux/memblock.h>
23 #include <linux/slab.h>
24 #include <linux/iommu.h>
25 #include <linux/io.h>
26 #include <linux/vmalloc.h>
27 #include <linux/sizes.h>
28
29 #include <asm/memory.h>
30 #include <asm/highmem.h>
31 #include <asm/cacheflush.h>
32 #include <asm/tlbflush.h>
33 #include <asm/mach/arch.h>
34 #include <asm/dma-iommu.h>
35 #include <asm/mach/map.h>
36 #include <asm/system_info.h>
37 #include <asm/dma-contiguous.h>
38
39 #include "mm.h"
40
41 /*
42  * The DMA API is built upon the notion of "buffer ownership".  A buffer
43  * is either exclusively owned by the CPU (and therefore may be accessed
44  * by it) or exclusively owned by the DMA device.  These helper functions
45  * represent the transitions between these two ownership states.
46  *
47  * Note, however, that on later ARMs, this notion does not work due to
48  * speculative prefetches.  We model our approach on the assumption that
49  * the CPU does do speculative prefetches, which means we clean caches
50  * before transfers and delay cache invalidation until transfer completion.
51  *
52  */
53 static void __dma_page_cpu_to_dev(struct page *, unsigned long,
54                 size_t, enum dma_data_direction);
55 static void __dma_page_dev_to_cpu(struct page *, unsigned long,
56                 size_t, enum dma_data_direction);
57
58 /**
59  * arm_dma_map_page - map a portion of a page for streaming DMA
60  * @dev: valid struct device pointer, or NULL for ISA and EISA-like devices
61  * @page: page that buffer resides in
62  * @offset: offset into page for start of buffer
63  * @size: size of buffer to map
64  * @dir: DMA transfer direction
65  *
66  * Ensure that any data held in the cache is appropriately discarded
67  * or written back.
68  *
69  * The device owns this memory once this call has completed.  The CPU
70  * can regain ownership by calling dma_unmap_page().
71  */
72 static dma_addr_t arm_dma_map_page(struct device *dev, struct page *page,
73              unsigned long offset, size_t size, enum dma_data_direction dir,
74              struct dma_attrs *attrs)
75 {
76         if (!dma_get_attr(DMA_ATTR_SKIP_CPU_SYNC, attrs))
77                 __dma_page_cpu_to_dev(page, offset, size, dir);
78         return pfn_to_dma(dev, page_to_pfn(page)) + offset;
79 }
80
81 static dma_addr_t arm_coherent_dma_map_page(struct device *dev, struct page *page,
82              unsigned long offset, size_t size, enum dma_data_direction dir,
83              struct dma_attrs *attrs)
84 {
85         return pfn_to_dma(dev, page_to_pfn(page)) + offset;
86 }
87
88 /**
89  * arm_dma_unmap_page - unmap a buffer previously mapped through dma_map_page()
90  * @dev: valid struct device pointer, or NULL for ISA and EISA-like devices
91  * @handle: DMA address of buffer
92  * @size: size of buffer (same as passed to dma_map_page)
93  * @dir: DMA transfer direction (same as passed to dma_map_page)
94  *
95  * Unmap a page streaming mode DMA translation.  The handle and size
96  * must match what was provided in the previous dma_map_page() call.
97  * All other usages are undefined.
98  *
99  * After this call, reads by the CPU to the buffer are guaranteed to see
100  * whatever the device wrote there.
101  */
102 static void arm_dma_unmap_page(struct device *dev, dma_addr_t handle,
103                 size_t size, enum dma_data_direction dir,
104                 struct dma_attrs *attrs)
105 {
106         if (!dma_get_attr(DMA_ATTR_SKIP_CPU_SYNC, attrs))
107                 __dma_page_dev_to_cpu(pfn_to_page(dma_to_pfn(dev, handle)),
108                                       handle & ~PAGE_MASK, size, dir);
109 }
110
111 static void arm_dma_sync_single_for_cpu(struct device *dev,
112                 dma_addr_t handle, size_t size, enum dma_data_direction dir)
113 {
114         unsigned int offset = handle & (PAGE_SIZE - 1);
115         struct page *page = pfn_to_page(dma_to_pfn(dev, handle-offset));
116         __dma_page_dev_to_cpu(page, offset, size, dir);
117 }
118
119 static void arm_dma_sync_single_for_device(struct device *dev,
120                 dma_addr_t handle, size_t size, enum dma_data_direction dir)
121 {
122         unsigned int offset = handle & (PAGE_SIZE - 1);
123         struct page *page = pfn_to_page(dma_to_pfn(dev, handle-offset));
124         __dma_page_cpu_to_dev(page, offset, size, dir);
125 }
126
127 struct dma_map_ops arm_dma_ops = {
128         .alloc                  = arm_dma_alloc,
129         .free                   = arm_dma_free,
130         .mmap                   = arm_dma_mmap,
131         .get_sgtable            = arm_dma_get_sgtable,
132         .map_page               = arm_dma_map_page,
133         .unmap_page             = arm_dma_unmap_page,
134         .map_sg                 = arm_dma_map_sg,
135         .unmap_sg               = arm_dma_unmap_sg,
136         .sync_single_for_cpu    = arm_dma_sync_single_for_cpu,
137         .sync_single_for_device = arm_dma_sync_single_for_device,
138         .sync_sg_for_cpu        = arm_dma_sync_sg_for_cpu,
139         .sync_sg_for_device     = arm_dma_sync_sg_for_device,
140         .set_dma_mask           = arm_dma_set_mask,
141 };
142 EXPORT_SYMBOL(arm_dma_ops);
143
144 static void *arm_coherent_dma_alloc(struct device *dev, size_t size,
145         dma_addr_t *handle, gfp_t gfp, struct dma_attrs *attrs);
146 static void arm_coherent_dma_free(struct device *dev, size_t size, void *cpu_addr,
147                                   dma_addr_t handle, struct dma_attrs *attrs);
148
149 struct dma_map_ops arm_coherent_dma_ops = {
150         .alloc                  = arm_coherent_dma_alloc,
151         .free                   = arm_coherent_dma_free,
152         .mmap                   = arm_dma_mmap,
153         .get_sgtable            = arm_dma_get_sgtable,
154         .map_page               = arm_coherent_dma_map_page,
155         .map_sg                 = arm_dma_map_sg,
156         .set_dma_mask           = arm_dma_set_mask,
157 };
158 EXPORT_SYMBOL(arm_coherent_dma_ops);
159
160 static u64 get_coherent_dma_mask(struct device *dev)
161 {
162         u64 mask = (u64)arm_dma_limit;
163
164         if (dev) {
165                 mask = dev->coherent_dma_mask;
166
167                 /*
168                  * Sanity check the DMA mask - it must be non-zero, and
169                  * must be able to be satisfied by a DMA allocation.
170                  */
171                 if (mask == 0) {
172                         dev_warn(dev, "coherent DMA mask is unset\n");
173                         return 0;
174                 }
175
176                 if ((~mask) & (u64)arm_dma_limit) {
177                         dev_warn(dev, "coherent DMA mask %#llx is smaller "
178                                  "than system GFP_DMA mask %#llx\n",
179                                  mask, (u64)arm_dma_limit);
180                         return 0;
181                 }
182         }
183
184         return mask;
185 }
186
187 static void __dma_clear_buffer(struct page *page, size_t size)
188 {
189         /*
190          * Ensure that the allocated pages are zeroed, and that any data
191          * lurking in the kernel direct-mapped region is invalidated.
192          */
193         if (PageHighMem(page)) {
194                 phys_addr_t base = __pfn_to_phys(page_to_pfn(page));
195                 phys_addr_t end = base + size;
196                 while (size > 0) {
197                         void *ptr = kmap_atomic(page);
198                         memset(ptr, 0, PAGE_SIZE);
199                         dmac_flush_range(ptr, ptr + PAGE_SIZE);
200                         kunmap_atomic(ptr);
201                         page++;
202                         size -= PAGE_SIZE;
203                 }
204                 outer_flush_range(base, end);
205         } else {
206                 void *ptr = page_address(page);
207                 memset(ptr, 0, size);
208                 dmac_flush_range(ptr, ptr + size);
209                 outer_flush_range(__pa(ptr), __pa(ptr) + size);
210         }
211 }
212
213 /*
214  * Allocate a DMA buffer for 'dev' of size 'size' using the
215  * specified gfp mask.  Note that 'size' must be page aligned.
216  */
217 static struct page *__dma_alloc_buffer(struct device *dev, size_t size, gfp_t gfp)
218 {
219         unsigned long order = get_order(size);
220         struct page *page, *p, *e;
221
222         page = alloc_pages(gfp, order);
223         if (!page)
224                 return NULL;
225
226         /*
227          * Now split the huge page and free the excess pages
228          */
229         split_page(page, order);
230         for (p = page + (size >> PAGE_SHIFT), e = page + (1 << order); p < e; p++)
231                 __free_page(p);
232
233         __dma_clear_buffer(page, size);
234
235         return page;
236 }
237
238 /*
239  * Free a DMA buffer.  'size' must be page aligned.
240  */
241 static void __dma_free_buffer(struct page *page, size_t size)
242 {
243         struct page *e = page + (size >> PAGE_SHIFT);
244
245         while (page < e) {
246                 __free_page(page);
247                 page++;
248         }
249 }
250
251 #ifdef CONFIG_MMU
252 #ifdef CONFIG_HUGETLB_PAGE
253 #error ARM Coherent DMA allocator does not (yet) support huge TLB
254 #endif
255
256 static void *__alloc_from_contiguous(struct device *dev, size_t size,
257                                      pgprot_t prot, struct page **ret_page,
258                                      const void *caller);
259
260 static void *__alloc_remap_buffer(struct device *dev, size_t size, gfp_t gfp,
261                                  pgprot_t prot, struct page **ret_page,
262                                  const void *caller);
263
264 static void *
265 __dma_alloc_remap(struct page *page, size_t size, gfp_t gfp, pgprot_t prot,
266         const void *caller)
267 {
268         struct vm_struct *area;
269         unsigned long addr;
270
271         /*
272          * DMA allocation can be mapped to user space, so lets
273          * set VM_USERMAP flags too.
274          */
275         area = get_vm_area_caller(size, VM_ARM_DMA_CONSISTENT | VM_USERMAP,
276                                   caller);
277         if (!area)
278                 return NULL;
279         addr = (unsigned long)area->addr;
280         area->phys_addr = __pfn_to_phys(page_to_pfn(page));
281
282         if (ioremap_page_range(addr, addr + size, area->phys_addr, prot)) {
283                 vunmap((void *)addr);
284                 return NULL;
285         }
286         return (void *)addr;
287 }
288
289 static void __dma_free_remap(void *cpu_addr, size_t size)
290 {
291         unsigned int flags = VM_ARM_DMA_CONSISTENT | VM_USERMAP;
292         struct vm_struct *area = find_vm_area(cpu_addr);
293         if (!area || (area->flags & flags) != flags) {
294                 WARN(1, "trying to free invalid coherent area: %p\n", cpu_addr);
295                 return;
296         }
297         unmap_kernel_range((unsigned long)cpu_addr, size);
298         vunmap(cpu_addr);
299 }
300
301 #define DEFAULT_DMA_COHERENT_POOL_SIZE  SZ_256K
302
303 struct dma_pool {
304         size_t size;
305         spinlock_t lock;
306         unsigned long *bitmap;
307         unsigned long nr_pages;
308         void *vaddr;
309         struct page **pages;
310 };
311
312 static struct dma_pool atomic_pool = {
313         .size = DEFAULT_DMA_COHERENT_POOL_SIZE,
314 };
315
316 static int __init early_coherent_pool(char *p)
317 {
318         atomic_pool.size = memparse(p, &p);
319         return 0;
320 }
321 early_param("coherent_pool", early_coherent_pool);
322
323 void __init init_dma_coherent_pool_size(unsigned long size)
324 {
325         /*
326          * Catch any attempt to set the pool size too late.
327          */
328         BUG_ON(atomic_pool.vaddr);
329
330         /*
331          * Set architecture specific coherent pool size only if
332          * it has not been changed by kernel command line parameter.
333          */
334         if (atomic_pool.size == DEFAULT_DMA_COHERENT_POOL_SIZE)
335                 atomic_pool.size = size;
336 }
337
338 /*
339  * Initialise the coherent pool for atomic allocations.
340  */
341 static int __init atomic_pool_init(void)
342 {
343         struct dma_pool *pool = &atomic_pool;
344         pgprot_t prot = pgprot_dmacoherent(pgprot_kernel);
345         gfp_t gfp = GFP_KERNEL | GFP_DMA;
346         unsigned long nr_pages = pool->size >> PAGE_SHIFT;
347         unsigned long *bitmap;
348         struct page *page;
349         struct page **pages;
350         void *ptr;
351         int bitmap_size = BITS_TO_LONGS(nr_pages) * sizeof(long);
352
353         bitmap = kzalloc(bitmap_size, GFP_KERNEL);
354         if (!bitmap)
355                 goto no_bitmap;
356
357         pages = kzalloc(nr_pages * sizeof(struct page *), GFP_KERNEL);
358         if (!pages)
359                 goto no_pages;
360
361         if (IS_ENABLED(CONFIG_CMA))
362                 ptr = __alloc_from_contiguous(NULL, pool->size, prot, &page,
363                                               atomic_pool_init);
364         else
365                 ptr = __alloc_remap_buffer(NULL, pool->size, gfp, prot, &page,
366                                            atomic_pool_init);
367         if (ptr) {
368                 int i;
369
370                 for (i = 0; i < nr_pages; i++)
371                         pages[i] = page + i;
372
373                 spin_lock_init(&pool->lock);
374                 pool->vaddr = ptr;
375                 pool->pages = pages;
376                 pool->bitmap = bitmap;
377                 pool->nr_pages = nr_pages;
378                 pr_info("DMA: preallocated %u KiB pool for atomic coherent allocations\n",
379                        (unsigned)pool->size / 1024);
380                 return 0;
381         }
382
383         kfree(pages);
384 no_pages:
385         kfree(bitmap);
386 no_bitmap:
387         pr_err("DMA: failed to allocate %u KiB pool for atomic coherent allocation\n",
388                (unsigned)pool->size / 1024);
389         return -ENOMEM;
390 }
391 /*
392  * CMA is activated by core_initcall, so we must be called after it.
393  */
394 postcore_initcall(atomic_pool_init);
395
396 struct dma_contig_early_reserve {
397         phys_addr_t base;
398         unsigned long size;
399 };
400
401 static struct dma_contig_early_reserve dma_mmu_remap[MAX_CMA_AREAS] __initdata;
402
403 static int dma_mmu_remap_num __initdata;
404
405 void __init dma_contiguous_early_fixup(phys_addr_t base, unsigned long size)
406 {
407         dma_mmu_remap[dma_mmu_remap_num].base = base;
408         dma_mmu_remap[dma_mmu_remap_num].size = size;
409         dma_mmu_remap_num++;
410 }
411
412 void __init dma_contiguous_remap(void)
413 {
414         int i;
415         for (i = 0; i < dma_mmu_remap_num; i++) {
416                 phys_addr_t start = dma_mmu_remap[i].base;
417                 phys_addr_t end = start + dma_mmu_remap[i].size;
418                 struct map_desc map;
419                 unsigned long addr;
420
421                 if (end > arm_lowmem_limit)
422                         end = arm_lowmem_limit;
423                 if (start >= end)
424                         continue;
425
426                 map.pfn = __phys_to_pfn(start);
427                 map.virtual = __phys_to_virt(start);
428                 map.length = end - start;
429                 map.type = MT_MEMORY_DMA_READY;
430
431                 /*
432                  * Clear previous low-memory mapping
433                  */
434                 for (addr = __phys_to_virt(start); addr < __phys_to_virt(end);
435                      addr += PMD_SIZE)
436                         pmd_clear(pmd_off_k(addr));
437
438                 iotable_init(&map, 1);
439         }
440 }
441
442 static int __dma_update_pte(pte_t *pte, pgtable_t token, unsigned long addr,
443                             void *data)
444 {
445         struct page *page = virt_to_page(addr);
446         pgprot_t prot = *(pgprot_t *)data;
447
448         set_pte_ext(pte, mk_pte(page, prot), 0);
449         return 0;
450 }
451
452 static void __dma_remap(struct page *page, size_t size, pgprot_t prot)
453 {
454         unsigned long start = (unsigned long) page_address(page);
455         unsigned end = start + size;
456
457         apply_to_page_range(&init_mm, start, size, __dma_update_pte, &prot);
458         dsb();
459         flush_tlb_kernel_range(start, end);
460 }
461
462 static void *__alloc_remap_buffer(struct device *dev, size_t size, gfp_t gfp,
463                                  pgprot_t prot, struct page **ret_page,
464                                  const void *caller)
465 {
466         struct page *page;
467         void *ptr;
468         page = __dma_alloc_buffer(dev, size, gfp);
469         if (!page)
470                 return NULL;
471
472         ptr = __dma_alloc_remap(page, size, gfp, prot, caller);
473         if (!ptr) {
474                 __dma_free_buffer(page, size);
475                 return NULL;
476         }
477
478         *ret_page = page;
479         return ptr;
480 }
481
482 static void *__alloc_from_pool(size_t size, struct page **ret_page)
483 {
484         struct dma_pool *pool = &atomic_pool;
485         unsigned int count = PAGE_ALIGN(size) >> PAGE_SHIFT;
486         unsigned int pageno;
487         unsigned long flags;
488         void *ptr = NULL;
489         unsigned long align_mask;
490
491         if (!pool->vaddr) {
492                 WARN(1, "coherent pool not initialised!\n");
493                 return NULL;
494         }
495
496         /*
497          * Align the region allocation - allocations from pool are rather
498          * small, so align them to their order in pages, minimum is a page
499          * size. This helps reduce fragmentation of the DMA space.
500          */
501         align_mask = (1 << get_order(size)) - 1;
502
503         spin_lock_irqsave(&pool->lock, flags);
504         pageno = bitmap_find_next_zero_area(pool->bitmap, pool->nr_pages,
505                                             0, count, align_mask);
506         if (pageno < pool->nr_pages) {
507                 bitmap_set(pool->bitmap, pageno, count);
508                 ptr = pool->vaddr + PAGE_SIZE * pageno;
509                 *ret_page = pool->pages[pageno];
510         } else {
511                 pr_err_once("ERROR: %u KiB atomic DMA coherent pool is too small!\n"
512                             "Please increase it with coherent_pool= kernel parameter!\n",
513                             (unsigned)pool->size / 1024);
514         }
515         spin_unlock_irqrestore(&pool->lock, flags);
516
517         return ptr;
518 }
519
520 static bool __in_atomic_pool(void *start, size_t size)
521 {
522         struct dma_pool *pool = &atomic_pool;
523         void *end = start + size;
524         void *pool_start = pool->vaddr;
525         void *pool_end = pool->vaddr + pool->size;
526
527         if (start < pool_start || start >= pool_end)
528                 return false;
529
530         if (end <= pool_end)
531                 return true;
532
533         WARN(1, "Wrong coherent size(%p-%p) from atomic pool(%p-%p)\n",
534              start, end - 1, pool_start, pool_end - 1);
535
536         return false;
537 }
538
539 static int __free_from_pool(void *start, size_t size)
540 {
541         struct dma_pool *pool = &atomic_pool;
542         unsigned long pageno, count;
543         unsigned long flags;
544
545         if (!__in_atomic_pool(start, size))
546                 return 0;
547
548         pageno = (start - pool->vaddr) >> PAGE_SHIFT;
549         count = size >> PAGE_SHIFT;
550
551         spin_lock_irqsave(&pool->lock, flags);
552         bitmap_clear(pool->bitmap, pageno, count);
553         spin_unlock_irqrestore(&pool->lock, flags);
554
555         return 1;
556 }
557
558 static void *__alloc_from_contiguous(struct device *dev, size_t size,
559                                      pgprot_t prot, struct page **ret_page,
560                                      const void *caller)
561 {
562         unsigned long order = get_order(size);
563         size_t count = size >> PAGE_SHIFT;
564         struct page *page;
565         void *ptr;
566
567         page = dma_alloc_from_contiguous(dev, count, order);
568         if (!page)
569                 return NULL;
570
571         __dma_clear_buffer(page, size);
572
573         if (PageHighMem(page)) {
574                 ptr = __dma_alloc_remap(page, size, GFP_KERNEL, prot, caller);
575                 if (!ptr) {
576                         dma_release_from_contiguous(dev, page, count);
577                         return NULL;
578                 }
579         } else {
580                 __dma_remap(page, size, prot);
581                 ptr = page_address(page);
582         }
583         *ret_page = page;
584         return ptr;
585 }
586
587 static void __free_from_contiguous(struct device *dev, struct page *page,
588                                    void *cpu_addr, size_t size)
589 {
590         if (PageHighMem(page))
591                 __dma_free_remap(cpu_addr, size);
592         else
593                 __dma_remap(page, size, pgprot_kernel);
594         dma_release_from_contiguous(dev, page, size >> PAGE_SHIFT);
595 }
596
597 static inline pgprot_t __get_dma_pgprot(struct dma_attrs *attrs, pgprot_t prot)
598 {
599         prot = dma_get_attr(DMA_ATTR_WRITE_COMBINE, attrs) ?
600                             pgprot_writecombine(prot) :
601                             pgprot_dmacoherent(prot);
602         return prot;
603 }
604
605 #define nommu() 0
606
607 #else   /* !CONFIG_MMU */
608
609 #define nommu() 1
610
611 #define __get_dma_pgprot(attrs, prot)   __pgprot(0)
612 #define __alloc_remap_buffer(dev, size, gfp, prot, ret, c)      NULL
613 #define __alloc_from_pool(size, ret_page)                       NULL
614 #define __alloc_from_contiguous(dev, size, prot, ret, c)        NULL
615 #define __free_from_pool(cpu_addr, size)                        0
616 #define __free_from_contiguous(dev, page, cpu_addr, size)       do { } while (0)
617 #define __dma_free_remap(cpu_addr, size)                        do { } while (0)
618
619 #endif  /* CONFIG_MMU */
620
621 static void *__alloc_simple_buffer(struct device *dev, size_t size, gfp_t gfp,
622                                    struct page **ret_page)
623 {
624         struct page *page;
625         page = __dma_alloc_buffer(dev, size, gfp);
626         if (!page)
627                 return NULL;
628
629         *ret_page = page;
630         return page_address(page);
631 }
632
633
634
635 static void *__dma_alloc(struct device *dev, size_t size, dma_addr_t *handle,
636                          gfp_t gfp, pgprot_t prot, bool is_coherent, const void *caller)
637 {
638         u64 mask = get_coherent_dma_mask(dev);
639         struct page *page = NULL;
640         void *addr;
641
642 #ifdef CONFIG_DMA_API_DEBUG
643         u64 limit = (mask + 1) & ~mask;
644         if (limit && size >= limit) {
645                 dev_warn(dev, "coherent allocation too big (requested %#x mask %#llx)\n",
646                         size, mask);
647                 return NULL;
648         }
649 #endif
650
651         if (!mask)
652                 return NULL;
653
654         if (mask < 0xffffffffULL)
655                 gfp |= GFP_DMA;
656
657         /*
658          * Following is a work-around (a.k.a. hack) to prevent pages
659          * with __GFP_COMP being passed to split_page() which cannot
660          * handle them.  The real problem is that this flag probably
661          * should be 0 on ARM as it is not supported on this
662          * platform; see CONFIG_HUGETLBFS.
663          */
664         gfp &= ~(__GFP_COMP);
665
666         *handle = DMA_ERROR_CODE;
667         size = PAGE_ALIGN(size);
668
669         if (is_coherent || nommu())
670                 addr = __alloc_simple_buffer(dev, size, gfp, &page);
671         else if (!(gfp & __GFP_WAIT))
672                 addr = __alloc_from_pool(size, &page);
673         else if (!IS_ENABLED(CONFIG_CMA))
674                 addr = __alloc_remap_buffer(dev, size, gfp, prot, &page, caller);
675         else
676                 addr = __alloc_from_contiguous(dev, size, prot, &page, caller);
677
678         if (addr)
679                 *handle = pfn_to_dma(dev, page_to_pfn(page));
680
681         return addr;
682 }
683
684 /*
685  * Allocate DMA-coherent memory space and return both the kernel remapped
686  * virtual and bus address for that space.
687  */
688 void *arm_dma_alloc(struct device *dev, size_t size, dma_addr_t *handle,
689                     gfp_t gfp, struct dma_attrs *attrs)
690 {
691         pgprot_t prot = __get_dma_pgprot(attrs, pgprot_kernel);
692         void *memory;
693
694         if (dma_alloc_from_coherent(dev, size, handle, &memory))
695                 return memory;
696
697         return __dma_alloc(dev, size, handle, gfp, prot, false,
698                            __builtin_return_address(0));
699 }
700
701 static void *arm_coherent_dma_alloc(struct device *dev, size_t size,
702         dma_addr_t *handle, gfp_t gfp, struct dma_attrs *attrs)
703 {
704         pgprot_t prot = __get_dma_pgprot(attrs, pgprot_kernel);
705         void *memory;
706
707         if (dma_alloc_from_coherent(dev, size, handle, &memory))
708                 return memory;
709
710         return __dma_alloc(dev, size, handle, gfp, prot, true,
711                            __builtin_return_address(0));
712 }
713
714 /*
715  * Create userspace mapping for the DMA-coherent memory.
716  */
717 int arm_dma_mmap(struct device *dev, struct vm_area_struct *vma,
718                  void *cpu_addr, dma_addr_t dma_addr, size_t size,
719                  struct dma_attrs *attrs)
720 {
721         int ret = -ENXIO;
722 #ifdef CONFIG_MMU
723         unsigned long nr_vma_pages = (vma->vm_end - vma->vm_start) >> PAGE_SHIFT;
724         unsigned long nr_pages = PAGE_ALIGN(size) >> PAGE_SHIFT;
725         unsigned long pfn = dma_to_pfn(dev, dma_addr);
726         unsigned long off = vma->vm_pgoff;
727
728         vma->vm_page_prot = __get_dma_pgprot(attrs, vma->vm_page_prot);
729
730         if (dma_mmap_from_coherent(dev, vma, cpu_addr, size, &ret))
731                 return ret;
732
733         if (off < nr_pages && nr_vma_pages <= (nr_pages - off)) {
734                 ret = remap_pfn_range(vma, vma->vm_start,
735                                       pfn + off,
736                                       vma->vm_end - vma->vm_start,
737                                       vma->vm_page_prot);
738         }
739 #endif  /* CONFIG_MMU */
740
741         return ret;
742 }
743
744 /*
745  * Free a buffer as defined by the above mapping.
746  */
747 static void __arm_dma_free(struct device *dev, size_t size, void *cpu_addr,
748                            dma_addr_t handle, struct dma_attrs *attrs,
749                            bool is_coherent)
750 {
751         struct page *page = pfn_to_page(dma_to_pfn(dev, handle));
752
753         if (dma_release_from_coherent(dev, get_order(size), cpu_addr))
754                 return;
755
756         size = PAGE_ALIGN(size);
757
758         if (is_coherent || nommu()) {
759                 __dma_free_buffer(page, size);
760         } else if (__free_from_pool(cpu_addr, size)) {
761                 return;
762         } else if (!IS_ENABLED(CONFIG_CMA)) {
763                 __dma_free_remap(cpu_addr, size);
764                 __dma_free_buffer(page, size);
765         } else {
766                 /*
767                  * Non-atomic allocations cannot be freed with IRQs disabled
768                  */
769                 WARN_ON(irqs_disabled());
770                 __free_from_contiguous(dev, page, cpu_addr, size);
771         }
772 }
773
774 void arm_dma_free(struct device *dev, size_t size, void *cpu_addr,
775                   dma_addr_t handle, struct dma_attrs *attrs)
776 {
777         __arm_dma_free(dev, size, cpu_addr, handle, attrs, false);
778 }
779
780 static void arm_coherent_dma_free(struct device *dev, size_t size, void *cpu_addr,
781                                   dma_addr_t handle, struct dma_attrs *attrs)
782 {
783         __arm_dma_free(dev, size, cpu_addr, handle, attrs, true);
784 }
785
786 int arm_dma_get_sgtable(struct device *dev, struct sg_table *sgt,
787                  void *cpu_addr, dma_addr_t handle, size_t size,
788                  struct dma_attrs *attrs)
789 {
790         struct page *page = pfn_to_page(dma_to_pfn(dev, handle));
791         int ret;
792
793         ret = sg_alloc_table(sgt, 1, GFP_KERNEL);
794         if (unlikely(ret))
795                 return ret;
796
797         sg_set_page(sgt->sgl, page, PAGE_ALIGN(size), 0);
798         return 0;
799 }
800
801 static void dma_cache_maint_page(struct page *page, unsigned long offset,
802         size_t size, enum dma_data_direction dir,
803         void (*op)(const void *, size_t, int))
804 {
805         unsigned long pfn;
806         size_t left = size;
807
808         pfn = page_to_pfn(page) + offset / PAGE_SIZE;
809         offset %= PAGE_SIZE;
810
811         /*
812          * A single sg entry may refer to multiple physically contiguous
813          * pages.  But we still need to process highmem pages individually.
814          * If highmem is not configured then the bulk of this loop gets
815          * optimized out.
816          */
817         do {
818                 size_t len = left;
819                 void *vaddr;
820
821                 page = pfn_to_page(pfn);
822
823                 if (PageHighMem(page)) {
824                         if (len + offset > PAGE_SIZE)
825                                 len = PAGE_SIZE - offset;
826
827                         if (cache_is_vipt_nonaliasing()) {
828                                 vaddr = kmap_atomic(page);
829                                 op(vaddr + offset, len, dir);
830                                 kunmap_atomic(vaddr);
831                         } else {
832                                 vaddr = kmap_high_get(page);
833                                 if (vaddr) {
834                                         op(vaddr + offset, len, dir);
835                                         kunmap_high(page);
836                                 }
837                         }
838                 } else {
839                         vaddr = page_address(page) + offset;
840                         op(vaddr, len, dir);
841                 }
842                 offset = 0;
843                 pfn++;
844                 left -= len;
845         } while (left);
846 }
847
848 /*
849  * Make an area consistent for devices.
850  * Note: Drivers should NOT use this function directly, as it will break
851  * platforms with CONFIG_DMABOUNCE.
852  * Use the driver DMA support - see dma-mapping.h (dma_sync_*)
853  */
854 static void __dma_page_cpu_to_dev(struct page *page, unsigned long off,
855         size_t size, enum dma_data_direction dir)
856 {
857         unsigned long paddr;
858
859         dma_cache_maint_page(page, off, size, dir, dmac_map_area);
860
861         paddr = page_to_phys(page) + off;
862         if (dir == DMA_FROM_DEVICE) {
863                 outer_inv_range(paddr, paddr + size);
864         } else {
865                 outer_clean_range(paddr, paddr + size);
866         }
867         /* FIXME: non-speculating: flush on bidirectional mappings? */
868 }
869
870 static void __dma_page_dev_to_cpu(struct page *page, unsigned long off,
871         size_t size, enum dma_data_direction dir)
872 {
873         unsigned long paddr = page_to_phys(page) + off;
874
875         /* FIXME: non-speculating: not required */
876         /* don't bother invalidating if DMA to device */
877         if (dir != DMA_TO_DEVICE)
878                 outer_inv_range(paddr, paddr + size);
879
880         dma_cache_maint_page(page, off, size, dir, dmac_unmap_area);
881
882         /*
883          * Mark the D-cache clean for this page to avoid extra flushing.
884          */
885         if (dir != DMA_TO_DEVICE && off == 0 && size >= PAGE_SIZE)
886                 set_bit(PG_dcache_clean, &page->flags);
887 }
888
889 /**
890  * arm_dma_map_sg - map a set of SG buffers for streaming mode DMA
891  * @dev: valid struct device pointer, or NULL for ISA and EISA-like devices
892  * @sg: list of buffers
893  * @nents: number of buffers to map
894  * @dir: DMA transfer direction
895  *
896  * Map a set of buffers described by scatterlist in streaming mode for DMA.
897  * This is the scatter-gather version of the dma_map_single interface.
898  * Here the scatter gather list elements are each tagged with the
899  * appropriate dma address and length.  They are obtained via
900  * sg_dma_{address,length}.
901  *
902  * Device ownership issues as mentioned for dma_map_single are the same
903  * here.
904  */
905 int arm_dma_map_sg(struct device *dev, struct scatterlist *sg, int nents,
906                 enum dma_data_direction dir, struct dma_attrs *attrs)
907 {
908         struct dma_map_ops *ops = get_dma_ops(dev);
909         struct scatterlist *s;
910         int i, j;
911
912         for_each_sg(sg, s, nents, i) {
913 #ifdef CONFIG_NEED_SG_DMA_LENGTH
914                 s->dma_length = s->length;
915 #endif
916                 s->dma_address = ops->map_page(dev, sg_page(s), s->offset,
917                                                 s->length, dir, attrs);
918                 if (dma_mapping_error(dev, s->dma_address))
919                         goto bad_mapping;
920         }
921         return nents;
922
923  bad_mapping:
924         for_each_sg(sg, s, i, j)
925                 ops->unmap_page(dev, sg_dma_address(s), sg_dma_len(s), dir, attrs);
926         return 0;
927 }
928
929 /**
930  * arm_dma_unmap_sg - unmap a set of SG buffers mapped by dma_map_sg
931  * @dev: valid struct device pointer, or NULL for ISA and EISA-like devices
932  * @sg: list of buffers
933  * @nents: number of buffers to unmap (same as was passed to dma_map_sg)
934  * @dir: DMA transfer direction (same as was passed to dma_map_sg)
935  *
936  * Unmap a set of streaming mode DMA translations.  Again, CPU access
937  * rules concerning calls here are the same as for dma_unmap_single().
938  */
939 void arm_dma_unmap_sg(struct device *dev, struct scatterlist *sg, int nents,
940                 enum dma_data_direction dir, struct dma_attrs *attrs)
941 {
942         struct dma_map_ops *ops = get_dma_ops(dev);
943         struct scatterlist *s;
944
945         int i;
946
947         for_each_sg(sg, s, nents, i)
948                 ops->unmap_page(dev, sg_dma_address(s), sg_dma_len(s), dir, attrs);
949 }
950
951 /**
952  * arm_dma_sync_sg_for_cpu
953  * @dev: valid struct device pointer, or NULL for ISA and EISA-like devices
954  * @sg: list of buffers
955  * @nents: number of buffers to map (returned from dma_map_sg)
956  * @dir: DMA transfer direction (same as was passed to dma_map_sg)
957  */
958 void arm_dma_sync_sg_for_cpu(struct device *dev, struct scatterlist *sg,
959                         int nents, enum dma_data_direction dir)
960 {
961         struct dma_map_ops *ops = get_dma_ops(dev);
962         struct scatterlist *s;
963         int i;
964
965         for_each_sg(sg, s, nents, i)
966                 ops->sync_single_for_cpu(dev, sg_dma_address(s), s->length,
967                                          dir);
968 }
969
970 /**
971  * arm_dma_sync_sg_for_device
972  * @dev: valid struct device pointer, or NULL for ISA and EISA-like devices
973  * @sg: list of buffers
974  * @nents: number of buffers to map (returned from dma_map_sg)
975  * @dir: DMA transfer direction (same as was passed to dma_map_sg)
976  */
977 void arm_dma_sync_sg_for_device(struct device *dev, struct scatterlist *sg,
978                         int nents, enum dma_data_direction dir)
979 {
980         struct dma_map_ops *ops = get_dma_ops(dev);
981         struct scatterlist *s;
982         int i;
983
984         for_each_sg(sg, s, nents, i)
985                 ops->sync_single_for_device(dev, sg_dma_address(s), s->length,
986                                             dir);
987 }
988
989 /*
990  * Return whether the given device DMA address mask can be supported
991  * properly.  For example, if your device can only drive the low 24-bits
992  * during bus mastering, then you would pass 0x00ffffff as the mask
993  * to this function.
994  */
995 int dma_supported(struct device *dev, u64 mask)
996 {
997         if (mask < (u64)arm_dma_limit)
998                 return 0;
999         return 1;
1000 }
1001 EXPORT_SYMBOL(dma_supported);
1002
1003 int arm_dma_set_mask(struct device *dev, u64 dma_mask)
1004 {
1005         if (!dev->dma_mask || !dma_supported(dev, dma_mask))
1006                 return -EIO;
1007
1008         *dev->dma_mask = dma_mask;
1009
1010         return 0;
1011 }
1012
1013 #define PREALLOC_DMA_DEBUG_ENTRIES      4096
1014
1015 static int __init dma_debug_do_init(void)
1016 {
1017         dma_debug_init(PREALLOC_DMA_DEBUG_ENTRIES);
1018         return 0;
1019 }
1020 fs_initcall(dma_debug_do_init);
1021
1022 #ifdef CONFIG_ARM_DMA_USE_IOMMU
1023
1024 /* IOMMU */
1025
1026 static unsigned int prefetch_page_count = 1;
1027 static unsigned int gap_page_count = 1;
1028
1029 #define PF_PAGES_SIZE (prefetch_page_count << PAGE_SHIFT)
1030 #define PG_PAGES (prefetch_page_count + gap_page_count)
1031
1032 static struct page *iova_gap_pages;
1033 static phys_addr_t iova_gap_phys;
1034
1035 static int __init iova_gap_pages_init(void)
1036 {
1037         unsigned long order = get_order(PF_PAGES_SIZE);
1038
1039         iova_gap_pages = alloc_pages(GFP_KERNEL, order);
1040         if (WARN_ON(!iova_gap_pages)) {
1041                 prefetch_page_count = 0;
1042                 return 0;
1043         }
1044
1045         __dma_clear_buffer(iova_gap_pages, PAGE_SIZE << order);
1046         iova_gap_phys = page_to_phys(iova_gap_pages);
1047         return 0;
1048 }
1049 core_initcall(iova_gap_pages_init);
1050
1051 static int pg_iommu_map(struct iommu_domain *domain, unsigned long iova,
1052                         phys_addr_t phys, size_t len, int prot)
1053 {
1054         int err;
1055
1056         err = iommu_map(domain, iova + len, iova_gap_phys,
1057                            PF_PAGES_SIZE, prot);
1058         if (err)
1059                 return err;
1060
1061         err = iommu_map(domain, iova, phys, len, prot);
1062         if (err)
1063                 iommu_unmap(domain, iova + len, PF_PAGES_SIZE);
1064
1065         return err;
1066 }
1067
1068 static size_t pg_iommu_unmap(struct iommu_domain *domain,
1069                              unsigned long iova, size_t len, int prot)
1070 {
1071         phys_addr_t phys_addr;
1072
1073         phys_addr = iommu_iova_to_phys(domain, iova + len);
1074         BUG_ON(phys_addr != iova_gap_phys);
1075         iommu_unmap(domain, iova + len, PF_PAGES_SIZE);
1076         return iommu_unmap(domain, iova, len);
1077 }
1078
1079 static int pg_iommu_map_pages(struct iommu_domain *domain, unsigned long iova,
1080                     struct page **pages, size_t count, int prot)
1081 {
1082         int err;
1083
1084         err = iommu_map(domain, iova + (count << PAGE_SHIFT),
1085                            iova_gap_phys,
1086                            PF_PAGES_SIZE, prot);
1087         if (err)
1088                 return err;
1089
1090         err = iommu_map_pages(domain, iova, pages, count, prot);
1091         if (err)
1092                 iommu_unmap(domain, iova + (count << PAGE_SHIFT), PF_PAGES_SIZE);
1093
1094         return err;
1095 }
1096
1097 static int pg_iommu_map_sg(struct iommu_domain *domain, unsigned long iova,
1098                  struct scatterlist *sgl, int nents, int prot)
1099 {
1100         int err;
1101
1102         err = iommu_map(domain, iova + (nents << PAGE_SHIFT),
1103                            iova_gap_phys,
1104                            PF_PAGES_SIZE, prot);
1105         if (err)
1106                 return err;
1107
1108         err = iommu_map_sg(domain, iova, sgl, nents, prot);
1109         if (err)
1110                 iommu_unmap(domain, iova + (nents << PAGE_SHIFT), PF_PAGES_SIZE);
1111         return err;
1112 }
1113
1114 static size_t arm_iommu_iova_get_free_total(struct device *dev)
1115 {
1116         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
1117         unsigned long flags;
1118         size_t size = 0;
1119         unsigned long start = 0;
1120
1121         BUG_ON(!dev);
1122         BUG_ON(!mapping);
1123
1124         spin_lock_irqsave(&mapping->lock, flags);
1125         while (1) {
1126                 unsigned long end;
1127
1128                 start = bitmap_find_next_zero_area(mapping->bitmap,
1129                                                    mapping->bits, start, 1, 0);
1130                 if (start > mapping->bits)
1131                         break;
1132
1133                 end = find_next_bit(mapping->bitmap, mapping->bits, start);
1134                 size += end - start;
1135                 start = end;
1136         }
1137         spin_unlock_irqrestore(&mapping->lock, flags);
1138         return size << (mapping->order + PAGE_SHIFT);
1139 }
1140
1141 static size_t arm_iommu_iova_get_free_max(struct device *dev)
1142 {
1143         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
1144         unsigned long flags;
1145         size_t max_free = 0;
1146         unsigned long start = 0;
1147
1148         spin_lock_irqsave(&mapping->lock, flags);
1149         while (1) {
1150                 unsigned long end;
1151
1152                 start = bitmap_find_next_zero_area(mapping->bitmap,
1153                                                    mapping->bits, start, 1, 0);
1154                 if (start > mapping->bits)
1155                         break;
1156
1157                 end = find_next_bit(mapping->bitmap, mapping->bits, start);
1158                 max_free = max_t(size_t, max_free, end - start);
1159                 start = end;
1160         }
1161         spin_unlock_irqrestore(&mapping->lock, flags);
1162         return max_free << (mapping->order + PAGE_SHIFT);
1163 }
1164
1165 static inline dma_addr_t __alloc_iova(struct dma_iommu_mapping *mapping,
1166                                       size_t size, struct dma_attrs *attr)
1167 {
1168         unsigned int order = get_order(size);
1169         unsigned int align = 0;
1170         unsigned int count, start;
1171         unsigned long flags;
1172
1173         if (order > CONFIG_ARM_DMA_IOMMU_ALIGNMENT)
1174                 order = CONFIG_ARM_DMA_IOMMU_ALIGNMENT;
1175
1176         count = ((PAGE_ALIGN(size) >> PAGE_SHIFT) + PG_PAGES +
1177                  (1 << mapping->order) - 1) >> mapping->order;
1178
1179         if (order > mapping->order)
1180                 align = (1 << (order - mapping->order)) - 1;
1181
1182         spin_lock_irqsave(&mapping->lock, flags);
1183         start = bitmap_find_next_zero_area(mapping->bitmap, mapping->bits, 0,
1184                                            count, align);
1185         if (start > mapping->bits) {
1186                 spin_unlock_irqrestore(&mapping->lock, flags);
1187                 return DMA_ERROR_CODE;
1188         }
1189
1190         bitmap_set(mapping->bitmap, start, count);
1191         spin_unlock_irqrestore(&mapping->lock, flags);
1192
1193         return mapping->base + (start << (mapping->order + PAGE_SHIFT));
1194 }
1195
1196 static dma_addr_t __alloc_iova_at(struct dma_iommu_mapping *mapping,
1197                                   dma_addr_t *iova, size_t size,
1198                                   struct dma_attrs *attrs)
1199 {
1200         unsigned int count, start, orig;
1201         unsigned long flags;
1202         size_t bytes;
1203
1204         count = ((PAGE_ALIGN(size) >> PAGE_SHIFT) + PG_PAGES +
1205                  (1 << mapping->order) - 1) >> mapping->order;
1206         bytes = count << (mapping->order + PAGE_SHIFT);
1207
1208         spin_lock_irqsave(&mapping->lock, flags);
1209
1210         if ((*iova < mapping->base) || (bytes > mapping->end - *iova)) {
1211                 *iova = -ENXIO;
1212                 goto err_out;
1213         }
1214
1215         orig = (*iova - mapping->base) >> (mapping->order + PAGE_SHIFT);
1216         start = bitmap_find_next_zero_area(mapping->bitmap, mapping->bits,
1217                                            orig, count, 0);
1218
1219         if ((start > mapping->bits) || (orig != start)) {
1220                 *iova = -EINVAL;
1221                 goto err_out;
1222         }
1223
1224         bitmap_set(mapping->bitmap, start, count);
1225         spin_unlock_irqrestore(&mapping->lock, flags);
1226
1227         return mapping->base + (start << (mapping->order + PAGE_SHIFT));
1228
1229 err_out:
1230         spin_unlock_irqrestore(&mapping->lock, flags);
1231         return DMA_ERROR_CODE;
1232 }
1233
1234 static dma_addr_t arm_iommu_iova_alloc_at(struct device *dev, dma_addr_t *iova,
1235                                           size_t size, struct dma_attrs *attrs)
1236 {
1237         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
1238
1239         return __alloc_iova_at(mapping, iova, size, attrs);
1240 }
1241
1242 static dma_addr_t arm_iommu_iova_alloc(struct device *dev, size_t size,
1243                                        struct dma_attrs *attrs)
1244 {
1245         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
1246
1247         return __alloc_iova(mapping, size, attrs);
1248 }
1249
1250 static inline void __free_iova(struct dma_iommu_mapping *mapping,
1251                                dma_addr_t addr, size_t size,
1252                                struct dma_attrs *attrs)
1253 {
1254         unsigned int start = (addr - mapping->base) >>
1255                              (mapping->order + PAGE_SHIFT);
1256         unsigned int count = ((size >> PAGE_SHIFT) + PG_PAGES +
1257                               (1 << mapping->order) - 1) >> mapping->order;
1258         unsigned long flags;
1259
1260         spin_lock_irqsave(&mapping->lock, flags);
1261         bitmap_clear(mapping->bitmap, start, count);
1262         spin_unlock_irqrestore(&mapping->lock, flags);
1263 }
1264
1265 static void arm_iommu_iova_free(struct device *dev, dma_addr_t addr,
1266                                 size_t size, struct dma_attrs *attrs)
1267 {
1268         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
1269
1270         __free_iova(mapping, addr, size, attrs);
1271 }
1272
1273 static struct page **__iommu_alloc_buffer(struct device *dev, size_t size,
1274                                           gfp_t gfp, struct dma_attrs *attrs)
1275 {
1276         struct page **pages;
1277         int count = size >> PAGE_SHIFT;
1278         int array_size = count * sizeof(struct page *);
1279         int i = 0;
1280
1281         if (array_size <= PAGE_SIZE)
1282                 pages = kzalloc(array_size, gfp);
1283         else
1284                 pages = vzalloc(array_size);
1285         if (!pages)
1286                 return NULL;
1287
1288         if (dma_get_attr(DMA_ATTR_FORCE_CONTIGUOUS, attrs))
1289         {
1290                 unsigned long order = get_order(size);
1291                 struct page *page;
1292
1293                 page = dma_alloc_from_contiguous(dev, count, order);
1294                 if (!page)
1295                         goto error;
1296
1297                 __dma_clear_buffer(page, size);
1298
1299                 for (i = 0; i < count; i++)
1300                         pages[i] = page + i;
1301
1302                 return pages;
1303         }
1304
1305         /*
1306          * IOMMU can map any pages, so himem can also be used here
1307          */
1308         gfp |= __GFP_NOWARN | __GFP_HIGHMEM;
1309
1310         while (count) {
1311                 int j, order = __fls(count);
1312
1313                 pages[i] = alloc_pages(gfp, order);
1314                 while (!pages[i] && order)
1315                         pages[i] = alloc_pages(gfp, --order);
1316                 if (!pages[i])
1317                         goto error;
1318
1319                 if (order) {
1320                         split_page(pages[i], order);
1321                         j = 1 << order;
1322                         while (--j)
1323                                 pages[i + j] = pages[i] + j;
1324                 }
1325
1326                 __dma_clear_buffer(pages[i], PAGE_SIZE << order);
1327                 i += 1 << order;
1328                 count -= 1 << order;
1329         }
1330
1331         return pages;
1332 error:
1333         while (i--)
1334                 if (pages[i])
1335                         __free_pages(pages[i], 0);
1336         if (array_size <= PAGE_SIZE)
1337                 kfree(pages);
1338         else
1339                 vfree(pages);
1340         return NULL;
1341 }
1342
1343 static int __iommu_free_buffer(struct device *dev, struct page **pages,
1344                                size_t size, struct dma_attrs *attrs)
1345 {
1346         int count = size >> PAGE_SHIFT;
1347         int array_size = count * sizeof(struct page *);
1348         int i;
1349
1350         if (dma_get_attr(DMA_ATTR_FORCE_CONTIGUOUS, attrs)) {
1351                 dma_release_from_contiguous(dev, pages[0], count);
1352         } else {
1353                 for (i = 0; i < count; i++)
1354                         if (pages[i])
1355                                 __free_pages(pages[i], 0);
1356         }
1357
1358         if (array_size <= PAGE_SIZE)
1359                 kfree(pages);
1360         else
1361                 vfree(pages);
1362         return 0;
1363 }
1364
1365 /*
1366  * Create a CPU mapping for a specified pages
1367  */
1368 static void *
1369 __iommu_alloc_remap(struct page **pages, size_t size, gfp_t gfp, pgprot_t prot,
1370                     const void *caller)
1371 {
1372         unsigned int i, nr_pages = PAGE_ALIGN(size) >> PAGE_SHIFT;
1373         struct vm_struct *area;
1374         unsigned long p;
1375
1376         area = get_vm_area_caller(size, VM_ARM_DMA_CONSISTENT | VM_USERMAP,
1377                                   caller);
1378         if (!area)
1379                 return NULL;
1380
1381         area->pages = pages;
1382         area->nr_pages = nr_pages;
1383         p = (unsigned long)area->addr;
1384
1385         for (i = 0; i < nr_pages; i++) {
1386                 phys_addr_t phys = __pfn_to_phys(page_to_pfn(pages[i]));
1387                 if (ioremap_page_range(p, p + PAGE_SIZE, phys, prot))
1388                         goto err;
1389                 p += PAGE_SIZE;
1390         }
1391         return area->addr;
1392 err:
1393         unmap_kernel_range((unsigned long)area->addr, size);
1394         vunmap(area->addr);
1395         return NULL;
1396 }
1397
1398 /*
1399  * Create a mapping in device IO address space for specified pages
1400  */
1401 static dma_addr_t
1402 ____iommu_create_mapping(struct device *dev, dma_addr_t *req,
1403                          struct page **pages, size_t size,
1404                          struct dma_attrs *attrs)
1405 {
1406         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
1407         unsigned int count = PAGE_ALIGN(size) >> PAGE_SHIFT;
1408         dma_addr_t dma_addr, iova;
1409         int i, ret = DMA_ERROR_CODE;
1410
1411         if (req)
1412                 dma_addr = __alloc_iova_at(mapping, req, size, attrs);
1413         else
1414                 dma_addr = __alloc_iova(mapping, size, attrs);
1415
1416         if (dma_addr == DMA_ERROR_CODE)
1417                 return dma_addr;
1418
1419         iova = dma_addr;
1420         for (i = 0; i < count; ) {
1421                 unsigned int next_pfn = page_to_pfn(pages[i]) + 1;
1422                 phys_addr_t phys = page_to_phys(pages[i]);
1423                 unsigned int len, j;
1424
1425                 for (j = i + 1; j < count; j++, next_pfn++)
1426                         if (page_to_pfn(pages[j]) != next_pfn)
1427                                 break;
1428
1429                 len = (j - i) << PAGE_SHIFT;
1430                 ret = pg_iommu_map(mapping->domain, iova, phys, len,
1431                                    (int)attrs);
1432                 if (ret < 0)
1433                         goto fail;
1434                 iova += len;
1435                 i = j;
1436         }
1437         return dma_addr;
1438 fail:
1439         pg_iommu_unmap(mapping->domain, dma_addr, iova-dma_addr, (int)attrs);
1440         __free_iova(mapping, dma_addr, size, attrs);
1441         return DMA_ERROR_CODE;
1442 }
1443
1444 static dma_addr_t
1445 __iommu_create_mapping(struct device *dev, struct page **pages, size_t size,
1446                        struct dma_attrs *attrs)
1447 {
1448         return ____iommu_create_mapping(dev, NULL, pages, size, attrs);
1449 }
1450
1451 static int __iommu_remove_mapping(struct device *dev, dma_addr_t iova,
1452                                   size_t size, struct dma_attrs *attrs)
1453 {
1454         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
1455
1456         /*
1457          * add optional in-page offset from iova to size and align
1458          * result to page size
1459          */
1460         size = PAGE_ALIGN((iova & ~PAGE_MASK) + size);
1461         iova &= PAGE_MASK;
1462
1463         pg_iommu_unmap(mapping->domain, iova, size, (int)attrs);
1464         __free_iova(mapping, iova, size, attrs);
1465         return 0;
1466 }
1467
1468 static struct page **__atomic_get_pages(void *addr)
1469 {
1470         struct dma_pool *pool = &atomic_pool;
1471         struct page **pages = pool->pages;
1472         int offs = (addr - pool->vaddr) >> PAGE_SHIFT;
1473
1474         return pages + offs;
1475 }
1476
1477 static struct page **__iommu_get_pages(void *cpu_addr, struct dma_attrs *attrs)
1478 {
1479         struct vm_struct *area;
1480
1481         if (__in_atomic_pool(cpu_addr, PAGE_SIZE))
1482                 return __atomic_get_pages(cpu_addr);
1483
1484         if (dma_get_attr(DMA_ATTR_NO_KERNEL_MAPPING, attrs))
1485                 return cpu_addr;
1486
1487         area = find_vm_area(cpu_addr);
1488         if (area && (area->flags & VM_ARM_DMA_CONSISTENT))
1489                 return area->pages;
1490         return NULL;
1491 }
1492
1493 static void *__iommu_alloc_atomic(struct device *dev, size_t size,
1494                                   dma_addr_t *handle, struct dma_attrs *attrs)
1495 {
1496         struct page *page;
1497         void *addr;
1498
1499         addr = __alloc_from_pool(size, &page);
1500         if (!addr)
1501                 return NULL;
1502
1503         *handle = __iommu_create_mapping(dev, &page, size, attrs);
1504         if (*handle == DMA_ERROR_CODE)
1505                 goto err_mapping;
1506
1507         dev_dbg(dev, "%s() %08x(%x)\n", __func__, *handle, size);
1508         return addr;
1509
1510 err_mapping:
1511         __free_from_pool(addr, size);
1512         return NULL;
1513 }
1514
1515 static void __iommu_free_atomic(struct device *dev, void *cpu_addr,
1516                                 dma_addr_t handle, size_t size,
1517                                 struct dma_attrs *attrs)
1518 {
1519         __iommu_remove_mapping(dev, handle, size, attrs);
1520         __free_from_pool(cpu_addr, size);
1521         dev_dbg(dev, "%s() %08x(%x)\n", __func__, handle, size);
1522 }
1523
1524 static void *arm_iommu_alloc_attrs(struct device *dev, size_t size,
1525             dma_addr_t *handle, gfp_t gfp, struct dma_attrs *attrs)
1526 {
1527         pgprot_t prot = __get_dma_pgprot(attrs, pgprot_kernel);
1528         struct page **pages;
1529         void *addr = NULL;
1530
1531         /* Following is a work-around (a.k.a. hack) to prevent pages
1532          * with __GFP_COMP being passed to split_page() which cannot
1533          * handle them.  The real problem is that this flag probably
1534          * should be 0 on ARM as it is not supported on this
1535          * platform--see CONFIG_HUGETLB_PAGE. */
1536         gfp &= ~(__GFP_COMP);
1537
1538         size = PAGE_ALIGN(size);
1539
1540         if (gfp & GFP_ATOMIC)
1541
1542                 return __iommu_alloc_atomic(dev, size, handle, attrs);
1543
1544         pages = __iommu_alloc_buffer(dev, size, gfp, attrs);
1545         if (!pages)
1546                 return NULL;
1547
1548         if (*handle == DMA_ERROR_CODE)
1549                 *handle = __iommu_create_mapping(dev, pages, size, attrs);
1550         else
1551                 *handle = ____iommu_create_mapping(dev, handle, pages, size,
1552                                                    attrs);
1553
1554         if (*handle == DMA_ERROR_CODE)
1555                 goto err_buffer;
1556
1557         if (dma_get_attr(DMA_ATTR_NO_KERNEL_MAPPING, attrs))
1558                 return pages;
1559
1560         addr = __iommu_alloc_remap(pages, size, gfp, prot,
1561                                    __builtin_return_address(0));
1562         if (!addr)
1563                 goto err_mapping;
1564
1565         return addr;
1566
1567 err_mapping:
1568         __iommu_remove_mapping(dev, *handle, size, attrs);
1569 err_buffer:
1570         __iommu_free_buffer(dev, pages, size, attrs);
1571         return NULL;
1572 }
1573
1574 static int arm_iommu_mmap_attrs(struct device *dev, struct vm_area_struct *vma,
1575                     void *cpu_addr, dma_addr_t dma_addr, size_t size,
1576                     struct dma_attrs *attrs)
1577 {
1578         unsigned long uaddr = vma->vm_start;
1579         unsigned long usize = vma->vm_end - vma->vm_start;
1580         struct page **pages = __iommu_get_pages(cpu_addr, attrs);
1581
1582         vma->vm_page_prot = __get_dma_pgprot(attrs, vma->vm_page_prot);
1583
1584         if (!pages)
1585                 return -ENXIO;
1586
1587         do {
1588                 int ret = vm_insert_page(vma, uaddr, *pages++);
1589                 if (ret) {
1590                         pr_err("Remapping memory failed: %d\n", ret);
1591                         return ret;
1592                 }
1593                 uaddr += PAGE_SIZE;
1594                 usize -= PAGE_SIZE;
1595         } while (usize > 0);
1596
1597         return 0;
1598 }
1599
1600 /*
1601  * free a page as defined by the above mapping.
1602  * Must not be called with IRQs disabled.
1603  */
1604 void arm_iommu_free_attrs(struct device *dev, size_t size, void *cpu_addr,
1605                           dma_addr_t handle, struct dma_attrs *attrs)
1606 {
1607         struct page **pages = __iommu_get_pages(cpu_addr, attrs);
1608         size = PAGE_ALIGN(size);
1609
1610         if (!pages) {
1611                 WARN(1, "trying to free invalid coherent area: %p\n", cpu_addr);
1612                 return;
1613         }
1614
1615         if (__in_atomic_pool(cpu_addr, size)) {
1616                 __iommu_free_atomic(dev, cpu_addr, handle, size, attrs);
1617                 return;
1618         }
1619
1620         if (!dma_get_attr(DMA_ATTR_NO_KERNEL_MAPPING, attrs)) {
1621                 unmap_kernel_range((unsigned long)cpu_addr, size);
1622                 vunmap(cpu_addr);
1623         }
1624
1625         __iommu_remove_mapping(dev, handle, size, attrs);
1626         __iommu_free_buffer(dev, pages, size, attrs);
1627 }
1628
1629 static int arm_iommu_get_sgtable(struct device *dev, struct sg_table *sgt,
1630                                  void *cpu_addr, dma_addr_t dma_addr,
1631                                  size_t size, struct dma_attrs *attrs)
1632 {
1633         unsigned int count = PAGE_ALIGN(size) >> PAGE_SHIFT;
1634         struct page **pages = __iommu_get_pages(cpu_addr, attrs);
1635
1636         if (!pages)
1637                 return -ENXIO;
1638
1639         return sg_alloc_table_from_pages(sgt, pages, count, 0, size,
1640                                          GFP_KERNEL);
1641 }
1642
1643 /*
1644  * Map a part of the scatter-gather list into contiguous io address space
1645  */
1646 static int __map_sg_chunk(struct device *dev, struct scatterlist *sg,
1647                           size_t size, dma_addr_t *handle,
1648                           enum dma_data_direction dir, struct dma_attrs *attrs,
1649                           bool is_coherent)
1650 {
1651         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
1652         dma_addr_t iova, iova_base;
1653         int ret = 0;
1654         unsigned int count;
1655         struct scatterlist *s;
1656
1657         size = PAGE_ALIGN(size);
1658         *handle = DMA_ERROR_CODE;
1659
1660         iova_base = iova = __alloc_iova(mapping, size, attrs);
1661         if (iova == DMA_ERROR_CODE)
1662                 return -ENOMEM;
1663
1664         if (is_coherent || dma_get_attr(DMA_ATTR_SKIP_CPU_SYNC, attrs))
1665                 goto skip_cmaint;
1666
1667         for (count = 0, s = sg; count < (size >> PAGE_SHIFT); s = sg_next(s)) {
1668                 unsigned int len = PAGE_ALIGN(s->offset + s->length);
1669
1670                 __dma_page_cpu_to_dev(sg_page(s), s->offset, s->length, dir);
1671
1672                 count += len >> PAGE_SHIFT;
1673                 iova += len;
1674         }
1675
1676 skip_cmaint:
1677         count = size >> PAGE_SHIFT;
1678         ret = pg_iommu_map_sg(mapping->domain, iova_base, sg, count,
1679                               (int)attrs);
1680         if (WARN_ON(ret < 0))
1681                 goto fail;
1682
1683         *handle = iova_base;
1684
1685         return 0;
1686 fail:
1687         pg_iommu_unmap(mapping->domain, iova_base, count * PAGE_SIZE,
1688                        (int)attrs);
1689         __free_iova(mapping, iova_base, size, attrs);
1690         return ret;
1691 }
1692
1693 static int __iommu_map_sg(struct device *dev, struct scatterlist *sg, int nents,
1694                      enum dma_data_direction dir, struct dma_attrs *attrs,
1695                      bool is_coherent)
1696 {
1697         struct scatterlist *s = sg, *dma = sg, *start = sg;
1698         int i, count = 0;
1699         unsigned int offset = s->offset;
1700         unsigned int size = s->offset + s->length;
1701         unsigned int max = dma_get_max_seg_size(dev);
1702
1703         for (i = 1; i < nents; i++) {
1704                 s = sg_next(s);
1705
1706                 s->dma_address = DMA_ERROR_CODE;
1707                 s->dma_length = 0;
1708
1709                 if (s->offset || (size & ~PAGE_MASK) || size + s->length > max) {
1710                         if (__map_sg_chunk(dev, start, size, &dma->dma_address,
1711                             dir, attrs, is_coherent) < 0)
1712                                 goto bad_mapping;
1713
1714                         dma->dma_address += offset;
1715                         dma->dma_length = size - offset;
1716
1717                         size = offset = s->offset;
1718                         start = s;
1719                         dma = sg_next(dma);
1720                         count += 1;
1721                 }
1722                 size += s->length;
1723         }
1724         if (__map_sg_chunk(dev, start, size, &dma->dma_address, dir, attrs,
1725                 is_coherent) < 0)
1726                 goto bad_mapping;
1727
1728         dma->dma_address += offset;
1729         dma->dma_length = size - offset;
1730
1731         return count+1;
1732
1733 bad_mapping:
1734         for_each_sg(sg, s, count, i)
1735                 __iommu_remove_mapping(dev, sg_dma_address(s), sg_dma_len(s),
1736                         attrs);
1737         return 0;
1738 }
1739
1740 /**
1741  * arm_coherent_iommu_map_sg - map a set of SG buffers for streaming mode DMA
1742  * @dev: valid struct device pointer
1743  * @sg: list of buffers
1744  * @nents: number of buffers to map
1745  * @dir: DMA transfer direction
1746  *
1747  * Map a set of i/o coherent buffers described by scatterlist in streaming
1748  * mode for DMA. The scatter gather list elements are merged together (if
1749  * possible) and tagged with the appropriate dma address and length. They are
1750  * obtained via sg_dma_{address,length}.
1751  */
1752 int arm_coherent_iommu_map_sg(struct device *dev, struct scatterlist *sg,
1753                 int nents, enum dma_data_direction dir, struct dma_attrs *attrs)
1754 {
1755         return __iommu_map_sg(dev, sg, nents, dir, attrs, true);
1756 }
1757
1758 /**
1759  * arm_iommu_map_sg - map a set of SG buffers for streaming mode DMA
1760  * @dev: valid struct device pointer
1761  * @sg: list of buffers
1762  * @nents: number of buffers to map
1763  * @dir: DMA transfer direction
1764  *
1765  * Map a set of buffers described by scatterlist in streaming mode for DMA.
1766  * The scatter gather list elements are merged together (if possible) and
1767  * tagged with the appropriate dma address and length. They are obtained via
1768  * sg_dma_{address,length}.
1769  */
1770 int arm_iommu_map_sg(struct device *dev, struct scatterlist *sg,
1771                 int nents, enum dma_data_direction dir, struct dma_attrs *attrs)
1772 {
1773         return __iommu_map_sg(dev, sg, nents, dir, attrs, false);
1774 }
1775
1776 static void __iommu_unmap_sg(struct device *dev, struct scatterlist *sg,
1777                 int nents, enum dma_data_direction dir, struct dma_attrs *attrs,
1778                 bool is_coherent)
1779 {
1780         struct scatterlist *s;
1781         int i;
1782
1783         for_each_sg(sg, s, nents, i) {
1784                 if (sg_dma_len(s))
1785                         __iommu_remove_mapping(dev, sg_dma_address(s),
1786                                                sg_dma_len(s), attrs);
1787                 if (!is_coherent &&
1788                     !dma_get_attr(DMA_ATTR_SKIP_CPU_SYNC, attrs))
1789                         __dma_page_dev_to_cpu(sg_page(s), s->offset,
1790                                               s->length, dir);
1791         }
1792 }
1793
1794 /**
1795  * arm_coherent_iommu_unmap_sg - unmap a set of SG buffers mapped by dma_map_sg
1796  * @dev: valid struct device pointer
1797  * @sg: list of buffers
1798  * @nents: number of buffers to unmap (same as was passed to dma_map_sg)
1799  * @dir: DMA transfer direction (same as was passed to dma_map_sg)
1800  *
1801  * Unmap a set of streaming mode DMA translations.  Again, CPU access
1802  * rules concerning calls here are the same as for dma_unmap_single().
1803  */
1804 void arm_coherent_iommu_unmap_sg(struct device *dev, struct scatterlist *sg,
1805                 int nents, enum dma_data_direction dir, struct dma_attrs *attrs)
1806 {
1807         __iommu_unmap_sg(dev, sg, nents, dir, attrs, true);
1808 }
1809
1810 /**
1811  * arm_iommu_unmap_sg - unmap a set of SG buffers mapped by dma_map_sg
1812  * @dev: valid struct device pointer
1813  * @sg: list of buffers
1814  * @nents: number of buffers to unmap (same as was passed to dma_map_sg)
1815  * @dir: DMA transfer direction (same as was passed to dma_map_sg)
1816  *
1817  * Unmap a set of streaming mode DMA translations.  Again, CPU access
1818  * rules concerning calls here are the same as for dma_unmap_single().
1819  */
1820 void arm_iommu_unmap_sg(struct device *dev, struct scatterlist *sg, int nents,
1821                         enum dma_data_direction dir, struct dma_attrs *attrs)
1822 {
1823         __iommu_unmap_sg(dev, sg, nents, dir, attrs, false);
1824 }
1825
1826 /**
1827  * arm_iommu_sync_sg_for_cpu
1828  * @dev: valid struct device pointer
1829  * @sg: list of buffers
1830  * @nents: number of buffers to map (returned from dma_map_sg)
1831  * @dir: DMA transfer direction (same as was passed to dma_map_sg)
1832  */
1833 void arm_iommu_sync_sg_for_cpu(struct device *dev, struct scatterlist *sg,
1834                         int nents, enum dma_data_direction dir)
1835 {
1836         struct scatterlist *s;
1837         int i;
1838
1839         for_each_sg(sg, s, nents, i)
1840                 __dma_page_dev_to_cpu(sg_page(s), s->offset, s->length, dir);
1841
1842 }
1843
1844 /**
1845  * arm_iommu_sync_sg_for_device
1846  * @dev: valid struct device pointer
1847  * @sg: list of buffers
1848  * @nents: number of buffers to map (returned from dma_map_sg)
1849  * @dir: DMA transfer direction (same as was passed to dma_map_sg)
1850  */
1851 void arm_iommu_sync_sg_for_device(struct device *dev, struct scatterlist *sg,
1852                         int nents, enum dma_data_direction dir)
1853 {
1854         struct scatterlist *s;
1855         int i;
1856
1857         for_each_sg(sg, s, nents, i)
1858                 __dma_page_cpu_to_dev(sg_page(s), s->offset, s->length, dir);
1859 }
1860
1861
1862 /**
1863  * arm_coherent_iommu_map_page
1864  * @dev: valid struct device pointer
1865  * @page: page that buffer resides in
1866  * @offset: offset into page for start of buffer
1867  * @size: size of buffer to map
1868  * @dir: DMA transfer direction
1869  *
1870  * Coherent IOMMU aware version of arm_dma_map_page()
1871  */
1872 static dma_addr_t arm_coherent_iommu_map_page(struct device *dev, struct page *page,
1873              unsigned long offset, size_t size, enum dma_data_direction dir,
1874              struct dma_attrs *attrs)
1875 {
1876         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
1877         dma_addr_t dma_addr;
1878         int ret, len = PAGE_ALIGN(size + offset);
1879
1880         dma_addr = __alloc_iova(mapping, len, attrs);
1881         if (dma_addr == DMA_ERROR_CODE)
1882                 return dma_addr;
1883
1884         ret = pg_iommu_map(mapping->domain, dma_addr,
1885                            page_to_phys(page), len, (int)attrs);
1886         if (ret < 0)
1887                 goto fail;
1888
1889         return dma_addr + offset;
1890 fail:
1891         __free_iova(mapping, dma_addr, len, attrs);
1892         return DMA_ERROR_CODE;
1893 }
1894
1895 /**
1896  * arm_iommu_map_page
1897  * @dev: valid struct device pointer
1898  * @page: page that buffer resides in
1899  * @offset: offset into page for start of buffer
1900  * @size: size of buffer to map
1901  * @dir: DMA transfer direction
1902  *
1903  * IOMMU aware version of arm_dma_map_page()
1904  */
1905 static dma_addr_t arm_iommu_map_page(struct device *dev, struct page *page,
1906              unsigned long offset, size_t size, enum dma_data_direction dir,
1907              struct dma_attrs *attrs)
1908 {
1909         if (!dma_get_attr(DMA_ATTR_SKIP_CPU_SYNC, attrs))
1910                 __dma_page_cpu_to_dev(page, offset, size, dir);
1911
1912         return arm_coherent_iommu_map_page(dev, page, offset, size, dir, attrs);
1913 }
1914
1915 static dma_addr_t arm_iommu_map_page_at(struct device *dev, struct page *page,
1916                  dma_addr_t dma_addr, unsigned long offset, size_t size,
1917                  enum dma_data_direction dir, struct dma_attrs *attrs)
1918 {
1919         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
1920         int ret, len = PAGE_ALIGN(size + offset);
1921
1922         if (!dma_get_attr(DMA_ATTR_SKIP_CPU_SYNC, attrs))
1923                 __dma_page_cpu_to_dev(page, offset, size, dir);
1924
1925         ret = pg_iommu_map(mapping->domain, dma_addr,
1926                            page_to_phys(page), len, (int)attrs);
1927         if (ret < 0)
1928                 return DMA_ERROR_CODE;
1929
1930         return dma_addr + offset;
1931 }
1932
1933 static dma_addr_t arm_iommu_map_pages(struct device *dev, struct page **pages,
1934                                   dma_addr_t dma_handle, size_t count,
1935                                   enum dma_data_direction dir,
1936                                   struct dma_attrs *attrs)
1937 {
1938         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
1939         int ret;
1940
1941         if (!dma_get_attr(DMA_ATTR_SKIP_CPU_SYNC, attrs)) {
1942                 int i;
1943
1944                 for (i = 0; i < count; i++)
1945                         __dma_page_cpu_to_dev(pages[i], 0, PAGE_SIZE, dir);
1946         }
1947
1948         ret = pg_iommu_map_pages(mapping->domain, dma_handle, pages, count,
1949                                  (int)attrs);
1950         if (ret < 0)
1951                 return DMA_ERROR_CODE;
1952
1953         return dma_handle;
1954 }
1955
1956
1957 /**
1958  * arm_coherent_iommu_unmap_page
1959  * @dev: valid struct device pointer
1960  * @handle: DMA address of buffer
1961  * @size: size of buffer (same as passed to dma_map_page)
1962  * @dir: DMA transfer direction (same as passed to dma_map_page)
1963  *
1964  * Coherent IOMMU aware version of arm_dma_unmap_page()
1965  */
1966 static void arm_coherent_iommu_unmap_page(struct device *dev, dma_addr_t handle,
1967                 size_t size, enum dma_data_direction dir,
1968                 struct dma_attrs *attrs)
1969 {
1970         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
1971         dma_addr_t iova = handle & PAGE_MASK;
1972         int offset = handle & ~PAGE_MASK;
1973         int len = PAGE_ALIGN(size + offset);
1974
1975         if (!iova)
1976                 return;
1977
1978         pg_iommu_unmap(mapping->domain, iova, len, (int)attrs);
1979         if (!dma_get_attr(DMA_ATTR_SKIP_FREE_IOVA, attrs))
1980                 __free_iova(mapping, iova, len, attrs);
1981 }
1982
1983 /**
1984  * arm_iommu_unmap_page
1985  * @dev: valid struct device pointer
1986  * @handle: DMA address of buffer
1987  * @size: size of buffer (same as passed to dma_map_page)
1988  * @dir: DMA transfer direction (same as passed to dma_map_page)
1989  *
1990  * IOMMU aware version of arm_dma_unmap_page()
1991  */
1992 static void arm_iommu_unmap_page(struct device *dev, dma_addr_t handle,
1993                 size_t size, enum dma_data_direction dir,
1994                 struct dma_attrs *attrs)
1995 {
1996         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
1997         dma_addr_t iova = handle & PAGE_MASK;
1998         struct page *page = phys_to_page(iommu_iova_to_phys(mapping->domain, iova));
1999         int offset = handle & ~PAGE_MASK;
2000         int len = PAGE_ALIGN(size + offset);
2001
2002         if (!iova)
2003                 return;
2004
2005         if (!dma_get_attr(DMA_ATTR_SKIP_CPU_SYNC, attrs))
2006                 __dma_page_dev_to_cpu(page, offset, size, dir);
2007
2008         pg_iommu_unmap(mapping->domain, iova, len, (int)attrs);
2009         if (!dma_get_attr(DMA_ATTR_SKIP_FREE_IOVA, attrs))
2010                 __free_iova(mapping, iova, len, attrs);
2011 }
2012
2013 static void arm_iommu_sync_single_for_cpu(struct device *dev,
2014                 dma_addr_t handle, size_t size, enum dma_data_direction dir)
2015 {
2016         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
2017         dma_addr_t iova = handle & PAGE_MASK;
2018         struct page *page = phys_to_page(iommu_iova_to_phys(mapping->domain, iova));
2019         unsigned int offset = handle & ~PAGE_MASK;
2020
2021         if (!iova)
2022                 return;
2023
2024         if (WARN_ON(!pfn_valid(page_to_pfn(page))))
2025                 return;
2026
2027         __dma_page_dev_to_cpu(page, offset, size, dir);
2028 }
2029
2030 static void arm_iommu_sync_single_for_device(struct device *dev,
2031                 dma_addr_t handle, size_t size, enum dma_data_direction dir)
2032 {
2033         struct dma_iommu_mapping *mapping = dev->archdata.mapping;
2034         dma_addr_t iova = handle & PAGE_MASK;
2035         struct page *page = phys_to_page(iommu_iova_to_phys(mapping->domain, iova));
2036         unsigned int offset = handle & ~PAGE_MASK;
2037
2038         if (!iova)
2039                 return;
2040
2041         if (WARN_ON(!pfn_valid(page_to_pfn(page))))
2042                 return;
2043
2044         __dma_page_cpu_to_dev(page, offset, size, dir);
2045 }
2046
2047 struct dma_map_ops iommu_ops = {
2048         .alloc          = arm_iommu_alloc_attrs,
2049         .free           = arm_iommu_free_attrs,
2050         .mmap           = arm_iommu_mmap_attrs,
2051         .get_sgtable    = arm_iommu_get_sgtable,
2052
2053         .map_page               = arm_iommu_map_page,
2054         .map_pages              = arm_iommu_map_pages,
2055         .map_page_at            = arm_iommu_map_page_at,
2056         .unmap_page             = arm_iommu_unmap_page,
2057         .sync_single_for_cpu    = arm_iommu_sync_single_for_cpu,
2058         .sync_single_for_device = arm_iommu_sync_single_for_device,
2059
2060         .map_sg                 = arm_iommu_map_sg,
2061         .unmap_sg               = arm_iommu_unmap_sg,
2062         .sync_sg_for_cpu        = arm_iommu_sync_sg_for_cpu,
2063         .sync_sg_for_device     = arm_iommu_sync_sg_for_device,
2064
2065         .set_dma_mask           = arm_dma_set_mask,
2066
2067         .iova_alloc             = arm_iommu_iova_alloc,
2068         .iova_alloc_at          = arm_iommu_iova_alloc_at,
2069         .iova_free              = arm_iommu_iova_free,
2070         .iova_get_free_total    = arm_iommu_iova_get_free_total,
2071         .iova_get_free_max      = arm_iommu_iova_get_free_max,
2072 };
2073
2074 struct dma_map_ops iommu_coherent_ops = {
2075         .alloc          = arm_iommu_alloc_attrs,
2076         .free           = arm_iommu_free_attrs,
2077         .mmap           = arm_iommu_mmap_attrs,
2078         .get_sgtable    = arm_iommu_get_sgtable,
2079
2080         .map_page       = arm_coherent_iommu_map_page,
2081         .unmap_page     = arm_coherent_iommu_unmap_page,
2082
2083         .map_sg         = arm_coherent_iommu_map_sg,
2084         .unmap_sg       = arm_coherent_iommu_unmap_sg,
2085
2086         .set_dma_mask   = arm_dma_set_mask,
2087 };
2088
2089 /**
2090  * arm_iommu_create_mapping
2091  * @bus: pointer to the bus holding the client device (for IOMMU calls)
2092  * @base: start address of the valid IO address space
2093  * @size: size of the valid IO address space
2094  * @order: accuracy of the IO addresses allocations
2095  *
2096  * Creates a mapping structure which holds information about used/unused
2097  * IO address ranges, which is required to perform memory allocation and
2098  * mapping with IOMMU aware functions.
2099  *
2100  * The client device need to be attached to the mapping with
2101  * arm_iommu_attach_device function.
2102  */
2103 struct dma_iommu_mapping *
2104 arm_iommu_create_mapping(struct bus_type *bus, dma_addr_t base, size_t size,
2105                          int order)
2106 {
2107         unsigned int count = size >> (PAGE_SHIFT + order);
2108         unsigned int bitmap_size = BITS_TO_LONGS(count) * sizeof(long);
2109         struct dma_iommu_mapping *mapping;
2110         int err = -ENOMEM;
2111
2112         if (!count)
2113                 return ERR_PTR(-EINVAL);
2114
2115         mapping = kzalloc(sizeof(struct dma_iommu_mapping), GFP_KERNEL);
2116         if (!mapping)
2117                 goto err;
2118
2119         mapping->bitmap = kzalloc(bitmap_size, GFP_KERNEL);
2120         if (!mapping->bitmap)
2121                 goto err2;
2122
2123         base = round_up(base, 1 << (order + PAGE_SHIFT));
2124         mapping->base = base;
2125         mapping->end = base + size;
2126         mapping->bits = BITS_PER_BYTE * bitmap_size;
2127         mapping->order = order;
2128         spin_lock_init(&mapping->lock);
2129
2130         mapping->domain = iommu_domain_alloc(bus);
2131         if (!mapping->domain)
2132                 goto err3;
2133
2134         kref_init(&mapping->kref);
2135         return mapping;
2136 err3:
2137         kfree(mapping->bitmap);
2138 err2:
2139         kfree(mapping);
2140 err:
2141         return ERR_PTR(err);
2142 }
2143 EXPORT_SYMBOL_GPL(arm_iommu_create_mapping);
2144
2145 static void release_iommu_mapping(struct kref *kref)
2146 {
2147         struct dma_iommu_mapping *mapping =
2148                 container_of(kref, struct dma_iommu_mapping, kref);
2149
2150         iommu_domain_free(mapping->domain);
2151         kfree(mapping->bitmap);
2152         kfree(mapping);
2153 }
2154
2155 void arm_iommu_release_mapping(struct dma_iommu_mapping *mapping)
2156 {
2157         if (mapping)
2158                 kref_put(&mapping->kref, release_iommu_mapping);
2159 }
2160 EXPORT_SYMBOL_GPL(arm_iommu_release_mapping);
2161
2162 /**
2163  * arm_iommu_attach_device
2164  * @dev: valid struct device pointer
2165  * @mapping: io address space mapping structure (returned from
2166  *      arm_iommu_create_mapping)
2167  *
2168  * Attaches specified io address space mapping to the provided device,
2169  * this replaces the dma operations (dma_map_ops pointer) with the
2170  * IOMMU aware version. More than one client might be attached to
2171  * the same io address space mapping.
2172  */
2173 int arm_iommu_attach_device(struct device *dev,
2174                             struct dma_iommu_mapping *mapping)
2175 {
2176         int err;
2177         struct dma_map_ops *org_ops;
2178         struct dma_iommu_mapping *org_map;
2179
2180         org_ops = get_dma_ops(dev);
2181         set_dma_ops(dev, &iommu_ops);
2182
2183         org_map = dev->archdata.mapping;
2184         dev->archdata.mapping = mapping;
2185
2186         err = iommu_attach_device(mapping->domain, dev);
2187         if (err) {
2188                 set_dma_ops(dev, org_ops);
2189                 dev->archdata.mapping = org_map;
2190                 return err;
2191         }
2192
2193         kref_get(&mapping->kref);
2194
2195         pr_debug("Attached IOMMU controller to %s device.\n", dev_name(dev));
2196         return 0;
2197 }
2198 EXPORT_SYMBOL_GPL(arm_iommu_attach_device);
2199
2200 /**
2201  * arm_iommu_detach_device
2202  * @dev: valid struct device pointer
2203  *
2204  * Detaches the provided device from a previously attached map.
2205  * This voids the dma operations (dma_map_ops pointer)
2206  */
2207 void arm_iommu_detach_device(struct device *dev)
2208 {
2209         struct dma_iommu_mapping *mapping;
2210
2211         mapping = to_dma_iommu_mapping(dev);
2212         if (!mapping) {
2213                 dev_warn(dev, "Not attached\n");
2214                 return;
2215         }
2216
2217         iommu_detach_device(mapping->domain, dev);
2218         kref_put(&mapping->kref, release_iommu_mapping);
2219         mapping = NULL;
2220         set_dma_ops(dev, NULL);
2221
2222         pr_debug("Detached IOMMU controller from %s device.\n", dev_name(dev));
2223 }
2224 EXPORT_SYMBOL_GPL(arm_iommu_detach_device);
2225
2226 #endif