148b66f8fSLars-Peter Clausen // SPDX-License-Identifier: GPL-2.0
248b66f8fSLars-Peter Clausen /*
348b66f8fSLars-Peter Clausen * Copyright 2017 Analog Devices Inc.
448b66f8fSLars-Peter Clausen * Author: Lars-Peter Clausen <lars@metafoo.de>
548b66f8fSLars-Peter Clausen */
648b66f8fSLars-Peter Clausen
748b66f8fSLars-Peter Clausen #include <linux/err.h>
848b66f8fSLars-Peter Clausen #include <linux/export.h>
948b66f8fSLars-Peter Clausen #include <linux/slab.h>
1048b66f8fSLars-Peter Clausen #include <linux/module.h>
1148b66f8fSLars-Peter Clausen
1248b66f8fSLars-Peter Clausen #include <linux/iio/iio.h>
1348b66f8fSLars-Peter Clausen #include <linux/iio/consumer.h>
1448b66f8fSLars-Peter Clausen #include <linux/iio/hw-consumer.h>
1548b66f8fSLars-Peter Clausen #include <linux/iio/buffer_impl.h>
1648b66f8fSLars-Peter Clausen
1748b66f8fSLars-Peter Clausen /**
1848b66f8fSLars-Peter Clausen * struct iio_hw_consumer - IIO hw consumer block
1948b66f8fSLars-Peter Clausen * @buffers: hardware buffers list head.
2048b66f8fSLars-Peter Clausen * @channels: IIO provider channels.
2148b66f8fSLars-Peter Clausen */
2248b66f8fSLars-Peter Clausen struct iio_hw_consumer {
2348b66f8fSLars-Peter Clausen struct list_head buffers;
2448b66f8fSLars-Peter Clausen struct iio_channel *channels;
2548b66f8fSLars-Peter Clausen };
2648b66f8fSLars-Peter Clausen
2748b66f8fSLars-Peter Clausen struct hw_consumer_buffer {
2848b66f8fSLars-Peter Clausen struct list_head head;
2948b66f8fSLars-Peter Clausen struct iio_dev *indio_dev;
3048b66f8fSLars-Peter Clausen struct iio_buffer buffer;
3148b66f8fSLars-Peter Clausen long scan_mask[];
3248b66f8fSLars-Peter Clausen };
3348b66f8fSLars-Peter Clausen
iio_buffer_to_hw_consumer_buffer(struct iio_buffer * buffer)3448b66f8fSLars-Peter Clausen static struct hw_consumer_buffer *iio_buffer_to_hw_consumer_buffer(
3548b66f8fSLars-Peter Clausen struct iio_buffer *buffer)
3648b66f8fSLars-Peter Clausen {
3748b66f8fSLars-Peter Clausen return container_of(buffer, struct hw_consumer_buffer, buffer);
3848b66f8fSLars-Peter Clausen }
3948b66f8fSLars-Peter Clausen
iio_hw_buf_release(struct iio_buffer * buffer)4048b66f8fSLars-Peter Clausen static void iio_hw_buf_release(struct iio_buffer *buffer)
4148b66f8fSLars-Peter Clausen {
4248b66f8fSLars-Peter Clausen struct hw_consumer_buffer *hw_buf =
4348b66f8fSLars-Peter Clausen iio_buffer_to_hw_consumer_buffer(buffer);
4448b66f8fSLars-Peter Clausen kfree(hw_buf);
4548b66f8fSLars-Peter Clausen }
4648b66f8fSLars-Peter Clausen
4748b66f8fSLars-Peter Clausen static const struct iio_buffer_access_funcs iio_hw_buf_access = {
4848b66f8fSLars-Peter Clausen .release = &iio_hw_buf_release,
4948b66f8fSLars-Peter Clausen .modes = INDIO_BUFFER_HARDWARE,
5048b66f8fSLars-Peter Clausen };
5148b66f8fSLars-Peter Clausen
iio_hw_consumer_get_buffer(struct iio_hw_consumer * hwc,struct iio_dev * indio_dev)5248b66f8fSLars-Peter Clausen static struct hw_consumer_buffer *iio_hw_consumer_get_buffer(
5348b66f8fSLars-Peter Clausen struct iio_hw_consumer *hwc, struct iio_dev *indio_dev)
5448b66f8fSLars-Peter Clausen {
5548b66f8fSLars-Peter Clausen struct hw_consumer_buffer *buf;
5648b66f8fSLars-Peter Clausen
5748b66f8fSLars-Peter Clausen list_for_each_entry(buf, &hwc->buffers, head) {
5848b66f8fSLars-Peter Clausen if (buf->indio_dev == indio_dev)
5948b66f8fSLars-Peter Clausen return buf;
6048b66f8fSLars-Peter Clausen }
6148b66f8fSLars-Peter Clausen
62*2d255ec5SGustavo A. R. Silva buf = kzalloc(struct_size(buf, scan_mask, BITS_TO_LONGS(indio_dev->masklength)),
63*2d255ec5SGustavo A. R. Silva GFP_KERNEL);
6448b66f8fSLars-Peter Clausen if (!buf)
6548b66f8fSLars-Peter Clausen return NULL;
6648b66f8fSLars-Peter Clausen
6748b66f8fSLars-Peter Clausen buf->buffer.access = &iio_hw_buf_access;
6848b66f8fSLars-Peter Clausen buf->indio_dev = indio_dev;
6948b66f8fSLars-Peter Clausen buf->buffer.scan_mask = buf->scan_mask;
7048b66f8fSLars-Peter Clausen
7148b66f8fSLars-Peter Clausen iio_buffer_init(&buf->buffer);
7248b66f8fSLars-Peter Clausen list_add_tail(&buf->head, &hwc->buffers);
7348b66f8fSLars-Peter Clausen
7448b66f8fSLars-Peter Clausen return buf;
7548b66f8fSLars-Peter Clausen }
7648b66f8fSLars-Peter Clausen
7748b66f8fSLars-Peter Clausen /**
7848b66f8fSLars-Peter Clausen * iio_hw_consumer_alloc() - Allocate IIO hardware consumer
7948b66f8fSLars-Peter Clausen * @dev: Pointer to consumer device.
8048b66f8fSLars-Peter Clausen *
8148b66f8fSLars-Peter Clausen * Returns a valid iio_hw_consumer on success or a ERR_PTR() on failure.
8248b66f8fSLars-Peter Clausen */
iio_hw_consumer_alloc(struct device * dev)8348b66f8fSLars-Peter Clausen struct iio_hw_consumer *iio_hw_consumer_alloc(struct device *dev)
8448b66f8fSLars-Peter Clausen {
8548b66f8fSLars-Peter Clausen struct hw_consumer_buffer *buf;
8648b66f8fSLars-Peter Clausen struct iio_hw_consumer *hwc;
8748b66f8fSLars-Peter Clausen struct iio_channel *chan;
8848b66f8fSLars-Peter Clausen int ret;
8948b66f8fSLars-Peter Clausen
9048b66f8fSLars-Peter Clausen hwc = kzalloc(sizeof(*hwc), GFP_KERNEL);
9148b66f8fSLars-Peter Clausen if (!hwc)
9248b66f8fSLars-Peter Clausen return ERR_PTR(-ENOMEM);
9348b66f8fSLars-Peter Clausen
9448b66f8fSLars-Peter Clausen INIT_LIST_HEAD(&hwc->buffers);
9548b66f8fSLars-Peter Clausen
9648b66f8fSLars-Peter Clausen hwc->channels = iio_channel_get_all(dev);
9748b66f8fSLars-Peter Clausen if (IS_ERR(hwc->channels)) {
9848b66f8fSLars-Peter Clausen ret = PTR_ERR(hwc->channels);
9948b66f8fSLars-Peter Clausen goto err_free_hwc;
10048b66f8fSLars-Peter Clausen }
10148b66f8fSLars-Peter Clausen
10248b66f8fSLars-Peter Clausen chan = &hwc->channels[0];
10348b66f8fSLars-Peter Clausen while (chan->indio_dev) {
10448b66f8fSLars-Peter Clausen buf = iio_hw_consumer_get_buffer(hwc, chan->indio_dev);
10548b66f8fSLars-Peter Clausen if (!buf) {
10648b66f8fSLars-Peter Clausen ret = -ENOMEM;
10748b66f8fSLars-Peter Clausen goto err_put_buffers;
10848b66f8fSLars-Peter Clausen }
10948b66f8fSLars-Peter Clausen set_bit(chan->channel->scan_index, buf->buffer.scan_mask);
11048b66f8fSLars-Peter Clausen chan++;
11148b66f8fSLars-Peter Clausen }
11248b66f8fSLars-Peter Clausen
11348b66f8fSLars-Peter Clausen return hwc;
11448b66f8fSLars-Peter Clausen
11548b66f8fSLars-Peter Clausen err_put_buffers:
11648b66f8fSLars-Peter Clausen list_for_each_entry(buf, &hwc->buffers, head)
11748b66f8fSLars-Peter Clausen iio_buffer_put(&buf->buffer);
11848b66f8fSLars-Peter Clausen iio_channel_release_all(hwc->channels);
11948b66f8fSLars-Peter Clausen err_free_hwc:
12048b66f8fSLars-Peter Clausen kfree(hwc);
12148b66f8fSLars-Peter Clausen return ERR_PTR(ret);
12248b66f8fSLars-Peter Clausen }
12348b66f8fSLars-Peter Clausen EXPORT_SYMBOL_GPL(iio_hw_consumer_alloc);
12448b66f8fSLars-Peter Clausen
12548b66f8fSLars-Peter Clausen /**
12648b66f8fSLars-Peter Clausen * iio_hw_consumer_free() - Free IIO hardware consumer
12748b66f8fSLars-Peter Clausen * @hwc: hw consumer to free.
12848b66f8fSLars-Peter Clausen */
iio_hw_consumer_free(struct iio_hw_consumer * hwc)12948b66f8fSLars-Peter Clausen void iio_hw_consumer_free(struct iio_hw_consumer *hwc)
13048b66f8fSLars-Peter Clausen {
13148b66f8fSLars-Peter Clausen struct hw_consumer_buffer *buf, *n;
13248b66f8fSLars-Peter Clausen
13348b66f8fSLars-Peter Clausen iio_channel_release_all(hwc->channels);
13448b66f8fSLars-Peter Clausen list_for_each_entry_safe(buf, n, &hwc->buffers, head)
13548b66f8fSLars-Peter Clausen iio_buffer_put(&buf->buffer);
13648b66f8fSLars-Peter Clausen kfree(hwc);
13748b66f8fSLars-Peter Clausen }
13848b66f8fSLars-Peter Clausen EXPORT_SYMBOL_GPL(iio_hw_consumer_free);
13948b66f8fSLars-Peter Clausen
devm_iio_hw_consumer_release(void * iio_hwc)140bfc1807aSYicong Yang static void devm_iio_hw_consumer_release(void *iio_hwc)
141b688c18dSArnaud Pouliquen {
142bfc1807aSYicong Yang iio_hw_consumer_free(iio_hwc);
143b688c18dSArnaud Pouliquen }
144b688c18dSArnaud Pouliquen
145b688c18dSArnaud Pouliquen /**
146b688c18dSArnaud Pouliquen * devm_iio_hw_consumer_alloc - Resource-managed iio_hw_consumer_alloc()
147b688c18dSArnaud Pouliquen * @dev: Pointer to consumer device.
148b688c18dSArnaud Pouliquen *
149b688c18dSArnaud Pouliquen * Managed iio_hw_consumer_alloc. iio_hw_consumer allocated with this function
150b688c18dSArnaud Pouliquen * is automatically freed on driver detach.
151b688c18dSArnaud Pouliquen *
152b688c18dSArnaud Pouliquen * returns pointer to allocated iio_hw_consumer on success, NULL on failure.
153b688c18dSArnaud Pouliquen */
devm_iio_hw_consumer_alloc(struct device * dev)154b688c18dSArnaud Pouliquen struct iio_hw_consumer *devm_iio_hw_consumer_alloc(struct device *dev)
155b688c18dSArnaud Pouliquen {
156bfc1807aSYicong Yang struct iio_hw_consumer *iio_hwc;
157bfc1807aSYicong Yang int ret;
158b688c18dSArnaud Pouliquen
159b688c18dSArnaud Pouliquen iio_hwc = iio_hw_consumer_alloc(dev);
160bfc1807aSYicong Yang if (IS_ERR(iio_hwc))
161bfc1807aSYicong Yang return iio_hwc;
162bfc1807aSYicong Yang
163bfc1807aSYicong Yang ret = devm_add_action_or_reset(dev, devm_iio_hw_consumer_release,
164bfc1807aSYicong Yang iio_hwc);
165bfc1807aSYicong Yang if (ret)
166bfc1807aSYicong Yang return ERR_PTR(ret);
167b688c18dSArnaud Pouliquen
168b688c18dSArnaud Pouliquen return iio_hwc;
169b688c18dSArnaud Pouliquen }
170b688c18dSArnaud Pouliquen EXPORT_SYMBOL_GPL(devm_iio_hw_consumer_alloc);
171b688c18dSArnaud Pouliquen
172b688c18dSArnaud Pouliquen /**
17348b66f8fSLars-Peter Clausen * iio_hw_consumer_enable() - Enable IIO hardware consumer
17448b66f8fSLars-Peter Clausen * @hwc: iio_hw_consumer to enable.
17548b66f8fSLars-Peter Clausen *
17648b66f8fSLars-Peter Clausen * Returns 0 on success.
17748b66f8fSLars-Peter Clausen */
iio_hw_consumer_enable(struct iio_hw_consumer * hwc)17848b66f8fSLars-Peter Clausen int iio_hw_consumer_enable(struct iio_hw_consumer *hwc)
17948b66f8fSLars-Peter Clausen {
18048b66f8fSLars-Peter Clausen struct hw_consumer_buffer *buf;
18148b66f8fSLars-Peter Clausen int ret;
18248b66f8fSLars-Peter Clausen
18348b66f8fSLars-Peter Clausen list_for_each_entry(buf, &hwc->buffers, head) {
18448b66f8fSLars-Peter Clausen ret = iio_update_buffers(buf->indio_dev, &buf->buffer, NULL);
18548b66f8fSLars-Peter Clausen if (ret)
18648b66f8fSLars-Peter Clausen goto err_disable_buffers;
18748b66f8fSLars-Peter Clausen }
18848b66f8fSLars-Peter Clausen
18948b66f8fSLars-Peter Clausen return 0;
19048b66f8fSLars-Peter Clausen
19148b66f8fSLars-Peter Clausen err_disable_buffers:
19248b66f8fSLars-Peter Clausen list_for_each_entry_continue_reverse(buf, &hwc->buffers, head)
19348b66f8fSLars-Peter Clausen iio_update_buffers(buf->indio_dev, NULL, &buf->buffer);
19448b66f8fSLars-Peter Clausen return ret;
19548b66f8fSLars-Peter Clausen }
19648b66f8fSLars-Peter Clausen EXPORT_SYMBOL_GPL(iio_hw_consumer_enable);
19748b66f8fSLars-Peter Clausen
19848b66f8fSLars-Peter Clausen /**
19948b66f8fSLars-Peter Clausen * iio_hw_consumer_disable() - Disable IIO hardware consumer
20048b66f8fSLars-Peter Clausen * @hwc: iio_hw_consumer to disable.
20148b66f8fSLars-Peter Clausen */
iio_hw_consumer_disable(struct iio_hw_consumer * hwc)20248b66f8fSLars-Peter Clausen void iio_hw_consumer_disable(struct iio_hw_consumer *hwc)
20348b66f8fSLars-Peter Clausen {
20448b66f8fSLars-Peter Clausen struct hw_consumer_buffer *buf;
20548b66f8fSLars-Peter Clausen
20648b66f8fSLars-Peter Clausen list_for_each_entry(buf, &hwc->buffers, head)
20748b66f8fSLars-Peter Clausen iio_update_buffers(buf->indio_dev, NULL, &buf->buffer);
20848b66f8fSLars-Peter Clausen }
20948b66f8fSLars-Peter Clausen EXPORT_SYMBOL_GPL(iio_hw_consumer_disable);
21048b66f8fSLars-Peter Clausen
21148b66f8fSLars-Peter Clausen MODULE_AUTHOR("Lars-Peter Clausen <lars@metafoo.de>");
21248b66f8fSLars-Peter Clausen MODULE_DESCRIPTION("Hardware consumer buffer the IIO framework");
21348b66f8fSLars-Peter Clausen MODULE_LICENSE("GPL v2");
214