Fix refcount bug in stealth mode

2 files changed, 2 insertions, 0 deletions

diff --git a/linux-core/drm_drv.c b/linux-core/drm_drv.c
index 81ec1c76..8ca3524c 100644
--- a/linux-core/drm_drv.c
+++ b/linux-core/drm_drv.c
@@ -369,6 +369,7 @@ int drm_init(struct pci_driver *driver,
 					       pid->subvendor, pid->subdevice,
 					       pdev))) {
 				/* stealth mode requires a manual probe */
+				pci_dev_get(pdev);
 				drm_probe(pdev, &pciidlist[i], driver_fn);
 			}
 		}
diff --git a/linux/drm_drv.h b/linux/drm_drv.h
index 00761eb3..6d698393 100644
--- a/linux/drm_drv.h
+++ b/linux/drm_drv.h
@@ -565,6 +565,7 @@ static int __init drm_init( void )
 			/* pass back in pdev to account for multiple identical cards */
 			while ((pdev = pci_get_subsys(pid->vendor, pid->device, pid->subvendor, pid->subdevice, pdev))) {
 				/* stealth mode requires a manual probe */
+				pci_dev_get(pdev);
 				DRM(probe)(pdev, &DRM(pciidlist[i]));
 			}
 		}