iio: light: tcs3414: convert probe to device-managed routines

This change converts the driver to use only device-managed init routines in
the probe function of the driver.

This way, we no longer need the tcs3414_remove() hook.
We still need to keep the i2c_set_clientdata() call, as that's being used
for the PM routines.

And lastly, a devm_add_action_or_reset() hook is added to call the
powerdown handler when the chip is uninitialized or the probe fails.

Signed-off-by: Alexandru Ardelean <aardelean@deviqon.com>
Link: https://lore.kernel.org/r/20210624080534.9209-1-aardelean@deviqon.com
Signed-off-by: Jonathan Cameron <Jonathan.Cameron@huawei.com>

diff --git a/drivers/iio/light/tcs3414.c b/drivers/iio/light/tcs3414.c
index 0593abd..b872221 100644 (file)
--- a/drivers/iio/light/tcs3414.c
+++ b/drivers/iio/light/tcs3414.c
@@ -267,6 +267,18 @@ static const struct iio_buffer_setup_ops tcs3414_buffer_setup_ops = {
        .predisable = tcs3414_buffer_predisable,
 };
 
+static int tcs3414_powerdown(struct tcs3414_data *data)
+{
+       return i2c_smbus_write_byte_data(data->client, TCS3414_CONTROL,
+               data->control & ~(TCS3414_CONTROL_POWER |
+               TCS3414_CONTROL_ADC_EN));
+}
+
+static void tcs3414_powerdown_cleanup(void *data)
+{
+       tcs3414_powerdown(data);
+}
+
 static int tcs3414_probe(struct i2c_client *client,
                           const struct i2c_device_id *id)
 {
@@ -309,6 +321,11 @@ static int tcs3414_probe(struct i2c_client *client,
        if (ret < 0)
                return ret;
 
+       ret = devm_add_action_or_reset(&client->dev, tcs3414_powerdown_cleanup,
+                                      data);
+       if (ret < 0)
+               return ret;
+
        data->timing = TCS3414_INTEG_12MS; /* free running */
        ret = i2c_smbus_write_byte_data(data->client, TCS3414_TIMING,
                data->timing);
@@ -320,38 +337,12 @@ static int tcs3414_probe(struct i2c_client *client,
                return ret;
        data->gain = ret;
 
-       ret = iio_triggered_buffer_setup(indio_dev, NULL,
+       ret = devm_iio_triggered_buffer_setup(&client->dev, indio_dev, NULL,
                tcs3414_trigger_handler, &tcs3414_buffer_setup_ops);
        if (ret < 0)
                return ret;
 
-       ret = iio_device_register(indio_dev);
-       if (ret < 0)
-               goto buffer_cleanup;
-
-       return 0;
-
-buffer_cleanup:
-       iio_triggered_buffer_cleanup(indio_dev);
-       return ret;
-}
-
-static int tcs3414_powerdown(struct tcs3414_data *data)
-{
-       return i2c_smbus_write_byte_data(data->client, TCS3414_CONTROL,
-               data->control & ~(TCS3414_CONTROL_POWER |
-               TCS3414_CONTROL_ADC_EN));
-}
-
-static int tcs3414_remove(struct i2c_client *client)
-{
-       struct iio_dev *indio_dev = i2c_get_clientdata(client);
-
-       iio_device_unregister(indio_dev);
-       iio_triggered_buffer_cleanup(indio_dev);
-       tcs3414_powerdown(iio_priv(indio_dev));
-
-       return 0;
+       return devm_iio_device_register(&client->dev, indio_dev);
 }
 
 #ifdef CONFIG_PM_SLEEP
@@ -385,7 +376,6 @@ static struct i2c_driver tcs3414_driver = {
                .pm     = &tcs3414_pm_ops,
        },
        .probe          = tcs3414_probe,
-       .remove         = tcs3414_remove,
        .id_table       = tcs3414_id,
 };
 module_i2c_driver(tcs3414_driver);