[<prev] [next>] [thread-next>] [day] [month] [year] [list]
Message-Id: <20200325090122.9308-1-butterflyhuangxx@gmail.com>
Date: Wed, 25 Mar 2020 17:01:22 +0800
From: Xiaolong Huang <butterflyhuangxx@...il.com>
To: mchehab@...nel.org, hverkuil-cisco@...all.nl, tglx@...utronix.de,
mpe@...erman.id.au, allison@...utok.net
Cc: linux-media@...r.kernel.org, linux-kernel@...r.kernel.org,
Xiaolong Huang <butterflyhuangxx@...il.com>
Subject: [PATCH] media: media/pci: prevent memory leak in bttv_probe
In bttv_probe if some functions such as pci_enable_device,
pci_set_dma_mask and request_mem_region fails the allocated
memory for btv should be released.
Signed-off-by: Xiaolong Huang <butterflyhuangxx@...il.com>
---
drivers/media/pci/bt8xx/bttv-driver.c | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/drivers/media/pci/bt8xx/bttv-driver.c b/drivers/media/pci/bt8xx/bttv-driver.c
index a359da7773a9..37ac59110383 100644
--- a/drivers/media/pci/bt8xx/bttv-driver.c
+++ b/drivers/media/pci/bt8xx/bttv-driver.c
@@ -4013,10 +4013,14 @@ static int bttv_probe(struct pci_dev *dev, const struct pci_device_id *pci_id)
btv->id = dev->device;
if (pci_enable_device(dev)) {
pr_warn("%d: Can't enable device\n", btv->c.nr);
+ bttvs[btv->c.nr] = NULL;
+ kfree(btv);
return -EIO;
}
if (pci_set_dma_mask(dev, DMA_BIT_MASK(32))) {
pr_warn("%d: No suitable DMA available\n", btv->c.nr);
+ bttvs[btv->c.nr] = NULL;
+ kfree(btv);
return -EIO;
}
if (!request_mem_region(pci_resource_start(dev,0),
@@ -4025,6 +4029,8 @@ static int bttv_probe(struct pci_dev *dev, const struct pci_device_id *pci_id)
pr_warn("%d: can't request iomem (0x%llx)\n",
btv->c.nr,
(unsigned long long)pci_resource_start(dev, 0));
+ bttvs[btv->c.nr] = NULL;
+ kfree(btv);
return -EBUSY;
}
pci_set_master(dev);
@@ -4211,6 +4217,8 @@ static int bttv_probe(struct pci_dev *dev, const struct pci_device_id *pci_id)
release_mem_region(pci_resource_start(btv->c.pci,0),
pci_resource_len(btv->c.pci,0));
pci_disable_device(btv->c.pci);
+ bttvs[btv->c.nr] = NULL;
+ kfree(btv);
return result;
}
--
2.17.1
Powered by blists - more mailing lists