[<prev] [next>] [thread-next>] [day] [month] [year] [list]
Message-ID: <92ff13341044a39953f0ba116cf7f9881488f8c8.1714034489.git.llfl@linux.alibaba.com>
Date: Thu, 25 Apr 2024 16:48:31 +0800
From: "Kun(llfl)" <llfl@...ux.alibaba.com>
To: Joerg Roedel <joro@...tes.org>,
Vasant Hegde <vasant.hegde@....com>
Cc: Suravee Suthikulpanit <suravee.suthikulpanit@....com>,
Will Deacon <will@...nel.org>,
Robin Murphy <robin.murphy@....com>,
iommu@...ts.linux.dev,
linux-kernel@...r.kernel.org
Subject: [PATCH 1/1] iommu/amd: Fix memory leak in alloc_pci_segment()
Fix the memory leak issue that occurs when resource allocation fails in
alloc_pci_segment(). The dev_table, alias_table, and rlookup_table were
introduced individually in commits 04230c119930, 99fc4ac3d297,
and eda797a27795, but they all fail to release allocated resources
when other allocations fail.
Fixes: 04230c119930 ("iommu/amd: Introduce per PCI segment device table")
Reported-by: Xuchun Shang <xuchun.shang@...ux.alibaba.com>
Signed-off-by: Kun(llfl) <llfl@...ux.alibaba.com>
---
drivers/iommu/amd/init.c | 15 ++++++++++++---
1 file changed, 12 insertions(+), 3 deletions(-)
diff --git a/drivers/iommu/amd/init.c b/drivers/iommu/amd/init.c
index ac6754a85f35..4ce567f39473 100644
--- a/drivers/iommu/amd/init.c
+++ b/drivers/iommu/amd/init.c
@@ -1642,13 +1642,22 @@ static struct amd_iommu_pci_seg *__init alloc_pci_segment(u16 id,
list_add_tail(&pci_seg->list, &amd_iommu_pci_seg_list);
if (alloc_dev_table(pci_seg))
- return NULL;
+ goto alloc_dev_fail;
if (alloc_alias_table(pci_seg))
- return NULL;
+ goto alloc_alias_fail;
if (alloc_rlookup_table(pci_seg))
- return NULL;
+ goto alloc_rlookup_fail;
return pci_seg;
+
+alloc_rlookup_fail:
+ free_rlookup_table(pci_seg);
+alloc_alias_fail:
+ free_alias_table(pci_seg);
+alloc_dev_fail:
+ free_dev_table(pci_seg);
+ kfree(pci_seg);
+ return NULL;
}
static struct amd_iommu_pci_seg *__init get_pci_segment(u16 id,
--
2.43.0
Powered by blists - more mailing lists