This is required so that we give up the last reference to the device.

Also, rework error path so that it is easier to read.

Signed-off-by: Levente Kurusa <le...@linux.com>
---
 arch/mips/txx9/generic/setup.c | 27 +++++++++++++++++----------
 1 file changed, 17 insertions(+), 10 deletions(-)

diff --git a/arch/mips/txx9/generic/setup.c b/arch/mips/txx9/generic/setup.c
index 2b0b83c..24332f5 100644
--- a/arch/mips/txx9/generic/setup.c
+++ b/arch/mips/txx9/generic/setup.c
@@ -937,6 +937,12 @@ static ssize_t txx9_sram_write(struct file *filp, struct 
kobject *kobj,
        return size;
 }
 
+void txx9_device_release(struct device *dev) {
+       struct txx9_sramc_dev *sramc_dev;
+       txx9_sramc_dev = container_of(dev, struct txx9_sramc_dev, dev);
+       kfree(txx9_sramc_dev);
+}
+
 void __init txx9_sramc_init(struct resource *r)
 {
        struct txx9_sramc_dev *dev;
@@ -951,8 +957,11 @@ void __init txx9_sramc_init(struct resource *r)
                return;
        size = resource_size(r);
        dev->base = ioremap(r->start, size);
-       if (!dev->base)
-               goto exit;
+       if (!dev->base) {
+               kfree(dev);
+               return;
+       }
+       dev->dev.release = &txx9_device_release;
        dev->dev.bus = &txx9_sramc_subsys;
        sysfs_bin_attr_init(&dev->bindata_attr);
        dev->bindata_attr.attr.name = "bindata";
@@ -963,17 +972,15 @@ void __init txx9_sramc_init(struct resource *r)
        dev->bindata_attr.private = dev;
        err = device_register(&dev->dev);
        if (err)
-               goto exit;
+               goto exit_put;
        err = sysfs_create_bin_file(&dev->dev.kobj, &dev->bindata_attr);
        if (err) {
                device_unregister(&dev->dev);
-               goto exit;
-       }
-       return;
-exit:
-       if (dev) {
-               if (dev->base)
-                       iounmap(dev->base);
+               iounmap(dev->base);
                kfree(dev);
        }
+       return;
+exit_put:
+       put_device(&dev->dev);
+       return;
 }
-- 
1.8.3.1

--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html
Please read the FAQ at  http://www.tux.org/lkml/

Reply via email to