iommu/amd: Handle large pages correctly in free_pagetable

commit 0b3fff54bc01e8e6064d222a33e6fa7adabd94cd upstream.

Make sure that we are skipping over large PTEs while walking
the page-table tree.

Fixes: 5c34c403b723 ("iommu/amd: Fix memory leak in free_pagetable")
Signed-off-by: Joerg Roedel <jroedel@suse.de>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>

1 files changed, 6 insertions, 0 deletions

diff --git a/drivers/iommu/amd_iommu.c b/drivers/iommu/amd_iommu.c
index e1c7e9e51045..ca9f4edbb940 100644
--- a/drivers/iommu/amd_iommu.c
+++ b/drivers/iommu/amd_iommu.c
@@ -1869,9 +1869,15 @@ static void free_pt_##LVL (unsigned long __pt)			\
         pt = (u64 *)__pt;                                       \
                                                                 \
         for (i = 0; i < 512; ++i) {                             \
+                /* PTE present? */                              \
                 if (!IOMMU_PTE_PRESENT(pt[i]))                  \
                         continue;                               \
                                                                 \
+                /* Large PTE? */                                \
+                if (PM_PTE_LEVEL(pt[i]) == 0 ||                 \
+                    PM_PTE_LEVEL(pt[i]) == 7)                   \
+                        continue;                               \
+                                                                \
                 p = (unsigned long)IOMMU_PTE_PAGE(pt[i]);       \
                 FN(p);                                          \
         }                                                       \