iommu/amd: Free IOMMU resources when disabled on command line

After we made sure that all IOMMUs have been disabled we
need to make sure that all resources we allocated are
released again.

Signed-off-by: Joerg Roedel <jroedel@suse.de>

1 files changed, 7 insertions, 0 deletions

diff --git a/drivers/iommu/amd_iommu_init.c b/drivers/iommu/amd_iommu_init.c
index 8cc507f96f3a..128f9665c326 100644
--- a/drivers/iommu/amd_iommu_init.c
+++ b/drivers/iommu/amd_iommu_init.c
@@ -2430,6 +2430,13 @@ static int __init state_next(void)
         case IOMMU_IVRS_DETECTED:
                 ret = early_amd_iommu_init();
                 init_state = ret ? IOMMU_INIT_ERROR : IOMMU_ACPI_FINISHED;
+                if (init_state == IOMMU_ACPI_FINISHED && amd_iommu_disabled) {
+                        pr_info("AMD-Vi: AMD IOMMU disabled on kernel command-line\n");
+                        free_dma_resources();
+                        free_iommu_resources();
+                        init_state = IOMMU_CMDLINE_DISABLED;
+                        ret = -EINVAL;
+                }
                 break;
         case IOMMU_ACPI_FINISHED:
                 early_enable_iommus();