*
  ****************************************************************************/
 
-static void free_page_list(struct page *freelist)
-{
-       while (freelist != NULL) {
-               unsigned long p = (unsigned long)page_address(freelist);
-
-               freelist = freelist->freelist;
-               free_page(p);
-       }
-}
-
-static struct page *free_pt_page(u64 *pt, struct page *freelist)
+static void free_pt_page(u64 *pt, struct list_head *freelist)
 {
        struct page *p = virt_to_page(pt);
 
-       p->freelist = freelist;
-
-       return p;
+       list_add_tail(&p->lru, freelist);
 }
 
-static struct page *free_pt_lvl(u64 *pt, struct page *freelist, int lvl)
+static void free_pt_lvl(u64 *pt, struct list_head *freelist, int lvl)
 {
        u64 *p;
        int i;
                 */
                p = IOMMU_PTE_PAGE(pt[i]);
                if (lvl > 2)
-                       freelist = free_pt_lvl(p, freelist, lvl - 1);
+                       free_pt_lvl(p, freelist, lvl - 1);
                else
-                       freelist = free_pt_page(p, freelist);
+                       free_pt_page(p, freelist);
        }
 
-       return free_pt_page(pt, freelist);
+       free_pt_page(pt, freelist);
 }
 
-static struct page *free_sub_pt(u64 *root, int mode, struct page *freelist)
+static void free_sub_pt(u64 *root, int mode, struct list_head *freelist)
 {
        switch (mode) {
        case PAGE_MODE_NONE:
        case PAGE_MODE_7_LEVEL:
                break;
        case PAGE_MODE_1_LEVEL:
-               freelist = free_pt_page(root, freelist);
+               free_pt_page(root, freelist);
                break;
        case PAGE_MODE_2_LEVEL:
        case PAGE_MODE_3_LEVEL:
        default:
                BUG();
        }
-
-       return freelist;
 }
 
 void amd_iommu_domain_set_pgtable(struct protection_domain *domain,
        return pte;
 }
 
-static struct page *free_clear_pte(u64 *pte, u64 pteval, struct page *freelist)
+static void free_clear_pte(u64 *pte, u64 pteval, struct list_head *freelist)
 {
        u64 *pt;
        int mode;
        }
 
        if (!IOMMU_PTE_PRESENT(pteval))
-               return freelist;
+               return;
 
        pt   = IOMMU_PTE_PAGE(pteval);
        mode = IOMMU_PTE_MODE(pteval);
 
-       return free_sub_pt(pt, mode, freelist);
+       free_sub_pt(pt, mode, freelist);
 }
 
 /*
                          phys_addr_t paddr, size_t size, int prot, gfp_t gfp)
 {
        struct protection_domain *dom = io_pgtable_ops_to_domain(ops);
-       struct page *freelist = NULL;
+       LIST_HEAD(freelist);
        bool updated = false;
        u64 __pte, *pte;
        int ret, i, count;
                goto out;
 
        for (i = 0; i < count; ++i)
-               freelist = free_clear_pte(&pte[i], pte[i], freelist);
+               free_clear_pte(&pte[i], pte[i], &freelist);
 
-       if (freelist != NULL)
+       if (!list_empty(&freelist))
                updated = true;
 
        if (count > 1) {
        }
 
        /* Everything flushed out, free pages now */
-       free_page_list(freelist);
+       put_pages_list(&freelist);
 
        return ret;
 }
 {
        struct amd_io_pgtable *pgtable = container_of(iop, struct amd_io_pgtable, iop);
        struct protection_domain *dom;
-       struct page *freelist = NULL;
+       LIST_HEAD(freelist);
 
        if (pgtable->mode == PAGE_MODE_NONE)
                return;
        BUG_ON(pgtable->mode < PAGE_MODE_NONE ||
               pgtable->mode > PAGE_MODE_6_LEVEL);
 
-       freelist = free_sub_pt(pgtable->root, pgtable->mode, freelist);
+       free_sub_pt(pgtable->root, pgtable->mode, &freelist);
 
-       free_page_list(freelist);
+       put_pages_list(&freelist);
 }
 
 static struct io_pgtable *v1_alloc_pgtable(struct io_pgtable_cfg *cfg, void *cookie)