From: Nikolay Balandin <[email protected]>

Signed-off-by: Nikolay Balandin <[email protected]>
---
 drivers/misc/eeprom/at24.c |   15 +++++----------
 drivers/misc/eeprom/at25.c |    5 ++---
 2 files changed, 7 insertions(+), 13 deletions(-)

diff --git a/drivers/misc/eeprom/at24.c b/drivers/misc/eeprom/at24.c
index 2baeec5..e110662 100644
--- a/drivers/misc/eeprom/at24.c
+++ b/drivers/misc/eeprom/at24.c
@@ -553,7 +553,7 @@ static int at24_probe(struct i2c_client *client, const 
struct i2c_device_id *id)
                num_addresses = DIV_ROUND_UP(chip.byte_len,
                        (chip.flags & AT24_FLAG_ADDR16) ? 65536 : 256);
 
-       at24 = kzalloc(sizeof(struct at24_data) +
+       at24 = devm_kzalloc(&client->dev, sizeof(struct at24_data) +
                num_addresses * sizeof(struct i2c_client *), GFP_KERNEL);
        if (!at24) {
                err = -ENOMEM;
@@ -596,10 +596,12 @@ static int at24_probe(struct i2c_client *client, const 
struct i2c_device_id *id)
                        at24->write_max = write_max;
 
                        /* buffer (data + address at the beginning) */
-                       at24->writebuf = kmalloc(write_max + 2, GFP_KERNEL);
+                       at24->writebuf = devm_kzalloc(&client->dev,
+                               write_max + 2, GFP_KERNEL);
+
                        if (!at24->writebuf) {
                                err = -ENOMEM;
-                               goto err_struct;
+                               goto err_out;
                        }
                } else {
                        dev_warn(&client->dev,
@@ -647,10 +649,6 @@ err_clients:
        for (i = 1; i < num_addresses; i++)
                if (at24->client[i])
                        i2c_unregister_device(at24->client[i]);
-
-       kfree(at24->writebuf);
-err_struct:
-       kfree(at24);
 err_out:
        dev_dbg(&client->dev, "probe error %d\n", err);
        return err;
@@ -666,9 +664,6 @@ static int at24_remove(struct i2c_client *client)
 
        for (i = 1; i < at24->num_addresses; i++)
                i2c_unregister_device(at24->client[i]);
-
-       kfree(at24->writebuf);
-       kfree(at24);
        return 0;
 }
 
diff --git a/drivers/misc/eeprom/at25.c b/drivers/misc/eeprom/at25.c
index ad8fd8e..f9d35c9 100644
--- a/drivers/misc/eeprom/at25.c
+++ b/drivers/misc/eeprom/at25.c
@@ -404,7 +404,8 @@ static int at25_probe(struct spi_device *spi)
                goto fail;
        }
 
-       if (!(at25 = kzalloc(sizeof *at25, GFP_KERNEL))) {
+       at25 = devm_kzalloc(&spi->dev, sizeof(struct at25_data), GFP_KERNEL);
+       if (!at25) {
                err = -ENOMEM;
                goto fail;
        }
@@ -455,7 +456,6 @@ static int at25_probe(struct spi_device *spi)
        return 0;
 fail:
        dev_dbg(&spi->dev, "probe err %d\n", err);
-       kfree(at25);
        return err;
 }
 
@@ -465,7 +465,6 @@ static int at25_remove(struct spi_device *spi)
 
        at25 = spi_get_drvdata(spi);
        sysfs_remove_bin_file(&spi->dev.kobj, &at25->bin);
-       kfree(at25);
        return 0;
 }
 
-- 
1.7.9.5

--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to [email protected]
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