- call device_unregister() only when device_register() is sucess.
- call put_device() when device_register() fails.
Signed-off-by: Rahul Ruikar<rahul.ruikar@xxxxxxxxx>
---
drivers/usb/gadget/langwell_udc.c | 18 ++++++++++++++----
drivers/usb/gadget/langwell_udc.h | 4 +++-
2 files changed, 17 insertions(+), 5 deletions(-)
diff --git a/drivers/usb/gadget/langwell_udc.c b/drivers/usb/gadget/langwell_udc.c
index c2d2a20..2b5ae16 100644
--- a/drivers/usb/gadget/langwell_udc.c
+++ b/drivers/usb/gadget/langwell_udc.c
@@ -2952,8 +2952,13 @@ static void langwell_udc_remove(struct pci_dev *pdev)
INFO(dev, "unbind\n");
DBG(dev, "<--- %s()\n", __func__);
- device_unregister(&dev->gadget.dev);
- device_remove_file(&pdev->dev,&dev_attr_langwell_udc);
+ if (dev->dev_reg_status == 2)
+ device_unregister(&dev->gadget.dev);
+ else if (dev->dev_reg_status == 1)
+ put_device(&dev->gadget.dev);
+
+ if (dev->dev_create_file)
+ device_remove_file(&pdev->dev,&dev_attr_langwell_udc);
#ifndef OTG_TRANSCEIVER
pci_set_drvdata(pdev, NULL);
@@ -3190,12 +3195,17 @@ static int langwell_udc_probe(struct pci_dev *pdev,
the_controller = dev;
retval = device_register(&dev->gadget.dev);
- if (retval)
+ if (retval) {
+ dev_reg_status = 1;
goto error;
+ }
+ dev_reg_status = 2;
retval = device_create_file(&pdev->dev,&dev_attr_langwell_udc);
- if (retval)
+ if (retval) {
+ dev_create_file = 1;