3 files changed, 60 insertions, 40 deletions

diff --git a/arch/powerpc/configs/skiroot_defconfig b/arch/powerpc/configs/skiroot_defconfig
index 5ba131c30f6b..1bcd468ab422 100644
--- a/arch/powerpc/configs/skiroot_defconfig
+++ b/arch/powerpc/configs/skiroot_defconfig
@@ -266,6 +266,7 @@ CONFIG_UDF_FS=m
 CONFIG_MSDOS_FS=m
 CONFIG_VFAT_FS=m
 CONFIG_PROC_KCORE=y
+CONFIG_HUGETLBFS=y
 # CONFIG_MISC_FILESYSTEMS is not set
 # CONFIG_NETWORK_FILESYSTEMS is not set
 CONFIG_NLS=y
diff --git a/arch/powerpc/mm/mmu_context_iommu.c b/arch/powerpc/mm/mmu_context_iommu.c
index e7a9c4f6bfca..8330f135294f 100644
--- a/arch/powerpc/mm/mmu_context_iommu.c
+++ b/arch/powerpc/mm/mmu_context_iommu.c
@@ -95,28 +95,15 @@ static long mm_iommu_do_alloc(struct mm_struct *mm, unsigned long ua,
                               unsigned long entries, unsigned long dev_hpa,
                               struct mm_iommu_table_group_mem_t **pmem)
 {
-        struct mm_iommu_table_group_mem_t *mem;
-        long i, ret, locked_entries = 0;
+        struct mm_iommu_table_group_mem_t *mem, *mem2;
+        long i, ret, locked_entries = 0, pinned = 0;
         unsigned int pageshift;
-
-        mutex_lock(&mem_list_mutex);
-
-        list_for_each_entry_rcu(mem, &mm->context.iommu_group_mem_list,
-                        next) {
-                /* Overlap? */
-                if ((mem->ua < (ua + (entries << PAGE_SHIFT))) &&
-                                (ua < (mem->ua +
-                                       (mem->entries << PAGE_SHIFT)))) {
-                        ret = -EINVAL;
-                        goto unlock_exit;
-                }
-
-        }
+        unsigned long entry, chunk;
 
         if (dev_hpa == MM_IOMMU_TABLE_INVALID_HPA) {
                 ret = mm_iommu_adjust_locked_vm(mm, entries, true);
                 if (ret)
-                        goto unlock_exit;
+                        return ret;
 
                 locked_entries = entries;
         }
@@ -148,17 +135,27 @@ static long mm_iommu_do_alloc(struct mm_struct *mm, unsigned long ua,
         }
 
         down_read(&mm->mmap_sem);
-        ret = get_user_pages_longterm(ua, entries, FOLL_WRITE, mem->hpages, NULL);
+        chunk = (1UL << (PAGE_SHIFT + MAX_ORDER - 1)) /
+                        sizeof(struct vm_area_struct *);
+        chunk = min(chunk, entries);
+        for (entry = 0; entry < entries; entry += chunk) {
+                unsigned long n = min(entries - entry, chunk);
+
+                ret = get_user_pages_longterm(ua + (entry << PAGE_SHIFT), n,
+                                FOLL_WRITE, mem->hpages + entry, NULL);
+                if (ret == n) {
+                        pinned += n;
+                        continue;
+                }
+                if (ret > 0)
+                        pinned += ret;
+                break;
+        }
         up_read(&mm->mmap_sem);
-        if (ret != entries) {
-                /* free the reference taken */
-                for (i = 0; i < ret; i++)
-                        put_page(mem->hpages[i]);
-
-                vfree(mem->hpas);
-                kfree(mem);
-                ret = -EFAULT;
-                goto unlock_exit;
+        if (pinned != entries) {
+                if (!ret)
+                        ret = -EFAULT;
+                goto free_exit;
         }
 
         pageshift = PAGE_SHIFT;
@@ -183,21 +180,43 @@ static long mm_iommu_do_alloc(struct mm_struct *mm, unsigned long ua,
         }
 
 good_exit:
-        ret = 0;
         atomic64_set(&mem->mapped, 1);
         mem->used = 1;
         mem->ua = ua;
         mem->entries = entries;
-        *pmem = mem;
 
-        list_add_rcu(&mem->next, &mm->context.iommu_group_mem_list);
+        mutex_lock(&mem_list_mutex);
 
-unlock_exit:
-        if (locked_entries && ret)
-                mm_iommu_adjust_locked_vm(mm, locked_entries, false);
+        list_for_each_entry_rcu(mem2, &mm->context.iommu_group_mem_list, next) {
+                /* Overlap? */
+                if ((mem2->ua < (ua + (entries << PAGE_SHIFT))) &&
+                                (ua < (mem2->ua +
+                                       (mem2->entries << PAGE_SHIFT)))) {
+                        ret = -EINVAL;
+                        mutex_unlock(&mem_list_mutex);
+                        goto free_exit;
+                }
+        }
+
+        list_add_rcu(&mem->next, &mm->context.iommu_group_mem_list);
 
         mutex_unlock(&mem_list_mutex);
 
+        *pmem = mem;
+
+        return 0;
+
+free_exit:
+        /* free the reference taken */
+        for (i = 0; i < pinned; i++)
+                put_page(mem->hpages[i]);
+
+        vfree(mem->hpas);
+        kfree(mem);
+
+unlock_exit:
+        mm_iommu_adjust_locked_vm(mm, locked_entries, false);
+
         return ret;
 }
 
@@ -266,7 +285,7 @@ static void mm_iommu_release(struct mm_iommu_table_group_mem_t *mem)
 long mm_iommu_put(struct mm_struct *mm, struct mm_iommu_table_group_mem_t *mem)
 {
         long ret = 0;
-        unsigned long entries, dev_hpa;
+        unsigned long unlock_entries = 0;
 
         mutex_lock(&mem_list_mutex);
 
@@ -287,17 +306,17 @@ long mm_iommu_put(struct mm_struct *mm, struct mm_iommu_table_group_mem_t *mem)
                 goto unlock_exit;
         }
 
+        if (mem->dev_hpa == MM_IOMMU_TABLE_INVALID_HPA)
+                unlock_entries = mem->entries;
+
         /* @mapped became 0 so now mappings are disabled, release the region */
-        entries = mem->entries;
-        dev_hpa = mem->dev_hpa;
         mm_iommu_release(mem);
 
-        if (dev_hpa == MM_IOMMU_TABLE_INVALID_HPA)
-                mm_iommu_adjust_locked_vm(mm, entries, false);
-
 unlock_exit:
         mutex_unlock(&mem_list_mutex);
 
+        mm_iommu_adjust_locked_vm(mm, unlock_entries, false);
+
         return ret;
 }
 EXPORT_SYMBOL_GPL(mm_iommu_put);
diff --git a/arch/powerpc/platforms/Kconfig.cputype b/arch/powerpc/platforms/Kconfig.cputype
index 842b2c7e156a..50cd09b4e05d 100644
--- a/arch/powerpc/platforms/Kconfig.cputype
+++ b/arch/powerpc/platforms/Kconfig.cputype
@@ -324,7 +324,7 @@ config ARCH_ENABLE_SPLIT_PMD_PTLOCK
 
 config PPC_RADIX_MMU
         bool "Radix MMU Support"
-        depends on PPC_BOOK3S_64
+        depends on PPC_BOOK3S_64 && HUGETLB_PAGE
         select ARCH_HAS_GIGANTIC_PAGE if (MEMORY_ISOLATION && COMPACTION) || CMA
         default y
         help