vfio-pci: Fix use after free

Reported by 0-day test infrastructure.

Fixes: ecaa1f6a0154 ("vfio-pci: Add VGA arbiter client")
Signed-off-by: Alex Williamson <alex.williamson@redhat.com>

diff --git a/drivers/vfio/pci/vfio_pci.c b/drivers/vfio/pci/vfio_pci.c
index 049b9e9e0a419047babf454e420272749afb204a..69fab0fd15aec4a5cb05c870984dc7e4bc9dcad4 100644 (file)
--- a/drivers/vfio/pci/vfio_pci.c
+++ b/drivers/vfio/pci/vfio_pci.c
@@ -953,6 +953,7 @@ static int vfio_pci_probe(struct pci_dev *pdev, const struct pci_device_id *id)
        if (ret) {
                iommu_group_put(group);
                kfree(vdev);
+               return ret;
        }
 
        if (vfio_pci_is_vga(pdev)) {