[PATCH 05/10] iommu/amd: Free page-table in protection_domain_free()

[Joerg Roedel]

From: Joerg Roedel <jroe...@suse.de>

Align release of the page-table with the place where it is allocated.

Signed-off-by: Joerg Roedel <jroe...@suse.de>
---
 drivers/iommu/amd/iommu.c | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/drivers/iommu/amd/iommu.c b/drivers/iommu/amd/iommu.c
index 0d5a5dbee9f3..5282ff6b8ea0 100644
--- a/drivers/iommu/amd/iommu.c
+++ b/drivers/iommu/amd/iommu.c
@@ -2387,12 +2387,18 @@ static void cleanup_domain(struct protection_domain 
*domain)
 
 static void protection_domain_free(struct protection_domain *domain)
 {
+       struct domain_pgtable pgtable;
+
        if (!domain)
                return;
 
        if (domain->id)
                domain_id_free(domain->id);
 
+       amd_iommu_domain_get_pgtable(domain, &pgtable);
+       atomic64_set(&domain->pt_root, 0);
+       free_pagetable(&pgtable);
+
        kfree(domain);
 }
 
@@ -2476,7 +2482,6 @@ static struct iommu_domain 
*amd_iommu_domain_alloc(unsigned type)
 static void amd_iommu_domain_free(struct iommu_domain *dom)
 {
        struct protection_domain *domain;
-       struct domain_pgtable pgtable;
 
        domain = to_pdomain(dom);
 
@@ -2494,10 +2499,6 @@ static void amd_iommu_domain_free(struct iommu_domain 
*dom)
                dma_ops_domain_free(domain);
                break;
        default:
-               amd_iommu_domain_get_pgtable(domain, &pgtable);
-               atomic64_set(&domain->pt_root, 0);
-               free_pagetable(&pgtable);
-
                if (domain->flags & PD_IOMMUV2_MASK)
                        free_gcr3_table(domain);
 
-- 
2.17.1

_______________________________________________
iommu mailing list
iommu@lists.linux-foundation.org
https://lists.linuxfoundation.org/mailman/listinfo/iommu