|
@@ -332,8 +332,10 @@ struct hnae_handle *hnae_get_handle(struct device *owner_dev,
|
|
return ERR_PTR(-ENODEV);
|
|
return ERR_PTR(-ENODEV);
|
|
|
|
|
|
handle = dev->ops->get_handle(dev, port_id);
|
|
handle = dev->ops->get_handle(dev, port_id);
|
|
- if (IS_ERR(handle))
|
|
|
|
|
|
+ if (IS_ERR(handle)) {
|
|
|
|
+ put_device(&dev->cls_dev);
|
|
return handle;
|
|
return handle;
|
|
|
|
+ }
|
|
|
|
|
|
handle->dev = dev;
|
|
handle->dev = dev;
|
|
handle->owner_dev = owner_dev;
|
|
handle->owner_dev = owner_dev;
|
|
@@ -356,6 +358,8 @@ out_when_init_queue:
|
|
for (j = i - 1; j >= 0; j--)
|
|
for (j = i - 1; j >= 0; j--)
|
|
hnae_fini_queue(handle->qs[j]);
|
|
hnae_fini_queue(handle->qs[j]);
|
|
|
|
|
|
|
|
+ put_device(&dev->cls_dev);
|
|
|
|
+
|
|
return ERR_PTR(-ENOMEM);
|
|
return ERR_PTR(-ENOMEM);
|
|
}
|
|
}
|
|
EXPORT_SYMBOL(hnae_get_handle);
|
|
EXPORT_SYMBOL(hnae_get_handle);
|
|
@@ -377,6 +381,8 @@ void hnae_put_handle(struct hnae_handle *h)
|
|
dev->ops->put_handle(h);
|
|
dev->ops->put_handle(h);
|
|
|
|
|
|
module_put(dev->owner);
|
|
module_put(dev->owner);
|
|
|
|
+
|
|
|
|
+ put_device(&dev->cls_dev);
|
|
}
|
|
}
|
|
EXPORT_SYMBOL(hnae_put_handle);
|
|
EXPORT_SYMBOL(hnae_put_handle);
|
|
|
|
|