[POWERPC] Make IOMMU code safe for > 132 GB of memory

Currently the IOMMU code allocates one page for the segment table, that
isn't safe if we have more than 132 GB of RAM.

Signed-off-by: Michael Ellerman <michael@ellerman.id.au>
Acked-by: Jeremy Kerr <jk@ozlabs.org>
Signed-off-by: Paul Mackerras <paulus@samba.org>

1 files changed, 4 insertions, 3 deletions

diff --git a/arch/powerpc/platforms/cell/iommu.c b/arch/powerpc/platforms/cell/iommu.c
index 7f45d59cdf2a..eb2a94b0dc4c 100644
--- a/arch/powerpc/platforms/cell/iommu.c
+++ b/arch/powerpc/platforms/cell/iommu.c
@@ -310,8 +310,8 @@ static void cell_iommu_setup_hardware(struct cbe_iommu *iommu, unsigned long siz
 {
         struct page *page;
         int ret, i;
-        unsigned long reg, segments, pages_per_segment, ptab_size, n_pte_pages;
-        unsigned long xlate_base;
+        unsigned long reg, segments, pages_per_segment, ptab_size, stab_size,
+                      n_pte_pages, xlate_base;
         unsigned int virq;
 
         if (cell_iommu_find_ioc(iommu->nid, &xlate_base))
@@ -328,7 +328,8 @@ static void cell_iommu_setup_hardware(struct cbe_iommu *iommu, unsigned long siz
                         __FUNCTION__, iommu->nid, segments, pages_per_segment);
 
         /* set up the segment table */
-        page = alloc_pages_node(iommu->nid, GFP_KERNEL, 0);
+        stab_size = segments * sizeof(unsigned long);
+        page = alloc_pages_node(iommu->nid, GFP_KERNEL, get_order(stab_size));
         BUG_ON(!page);
         iommu->stab = page_address(page);
         clear_page(iommu->stab);