iommu/vt-d: Free resources if failed to create domain for PCIe endpoint
Enhance function get_domain_for_dev() to release allocated resources if failed to create domain for PCIe endpoint, otherwise the allocated resources will get lost. Signed-off-by: Jiang Liu <jiang.liu@linux.intel.com> Signed-off-by: Joerg Roedel <joro@8bytes.org>
This commit is contained in:
@@ -2015,7 +2015,7 @@ static int dmar_insert_dev_info(int segment, int bus, int devfn,
|
|||||||
/* domain is initialized */
|
/* domain is initialized */
|
||||||
static struct dmar_domain *get_domain_for_dev(struct pci_dev *pdev, int gaw)
|
static struct dmar_domain *get_domain_for_dev(struct pci_dev *pdev, int gaw)
|
||||||
{
|
{
|
||||||
struct dmar_domain *domain;
|
struct dmar_domain *domain, *free = NULL;
|
||||||
struct intel_iommu *iommu;
|
struct intel_iommu *iommu;
|
||||||
struct dmar_drhd_unit *drhd;
|
struct dmar_drhd_unit *drhd;
|
||||||
struct pci_dev *dev_tmp;
|
struct pci_dev *dev_tmp;
|
||||||
@@ -2062,17 +2062,16 @@ static struct dmar_domain *get_domain_for_dev(struct pci_dev *pdev, int gaw)
|
|||||||
free_domain_mem(domain);
|
free_domain_mem(domain);
|
||||||
goto error;
|
goto error;
|
||||||
}
|
}
|
||||||
if (domain_init(domain, gaw)) {
|
free = domain;
|
||||||
domain_exit(domain);
|
if (domain_init(domain, gaw))
|
||||||
goto error;
|
goto error;
|
||||||
}
|
|
||||||
|
|
||||||
/* register pcie-to-pci device */
|
/* register pcie-to-pci device */
|
||||||
if (dev_tmp) {
|
if (dev_tmp) {
|
||||||
if (dmar_insert_dev_info(segment, bus, devfn, NULL, &domain)) {
|
if (dmar_insert_dev_info(segment, bus, devfn, NULL, &domain))
|
||||||
domain_exit(domain);
|
|
||||||
goto error;
|
goto error;
|
||||||
}
|
else
|
||||||
|
free = NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
found_domain:
|
found_domain:
|
||||||
@@ -2080,6 +2079,8 @@ found_domain:
|
|||||||
pdev, &domain) == 0)
|
pdev, &domain) == 0)
|
||||||
return domain;
|
return domain;
|
||||||
error:
|
error:
|
||||||
|
if (free)
|
||||||
|
domain_exit(free);
|
||||||
/* recheck it here, maybe others set it */
|
/* recheck it here, maybe others set it */
|
||||||
return find_domain(pdev);
|
return find_domain(pdev);
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user