[PATCH] iio: adc: ad7606: Fix trigger reference leak in ad7606_probe()

[Wentao Liang]

The reference taken with iio_trigger_get() is only dropped by
iio_dev_release() when INDIO_BUFFER_TRIGGERED is set. If
devm_request_threaded_irq() or triggered buffer setup fails, probe
returns before the mode is set and the trigger reference leaks. Move
the iio_trigger_get() after those setup steps succeed.

Fixes: cc49bd1652a4 ("staging: iio: adc: ad7606: Add support for threaded irq")
Cc: stable@xxxxxxxxxxxxxxx
Signed-off-by: Wentao Liang <vulab@xxxxxxxxxxx>
---
 drivers/iio/adc/ad7606.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/drivers/iio/adc/ad7606.c b/drivers/iio/adc/ad7606.c
index d9271894f091..2deef474001b 100644
--- a/drivers/iio/adc/ad7606.c
+++ b/drivers/iio/adc/ad7606.c
@@ -1638,8 +1638,6 @@ int ad7606_probe(struct device *dev, int irq, void __iomem *base_address,
 		if (ret)
 			return ret;
 
-		indio_dev->trig = iio_trigger_get(st->trig);
-
 		ret = devm_request_threaded_irq(dev, irq, NULL, &ad7606_interrupt,
 						IRQF_TRIGGER_FALLING | IRQF_ONESHOT,
 						chip_info->name, indio_dev);
@@ -1652,6 +1650,8 @@ int ad7606_probe(struct device *dev, int irq, void __iomem *base_address,
 						      &ad7606_buffer_ops);
 		if (ret)
 			return ret;
+
+		indio_dev->trig = iio_trigger_get(st->trig);
 	}
 
 	st->write_scale = ad7606_write_scale_hw;
-- 
2.34.1