[mirror_ubuntu-jammy-kernel.git] / drivers / iio / buffer / industrialio-buffer-cb.c

/* The industrial I/O callback buffer
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 as published by
 * the Free Software Foundation.
 */

#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/slab.h>
#include <linux/err.h>
#include <linux/export.h>
#include <linux/iio/iio.h>
#include <linux/iio/buffer_impl.h>
#include <linux/iio/consumer.h>

struct iio_cb_buffer {
	struct iio_buffer buffer;
	int (*cb)(const void *data, void *private);
	void *private;
	struct iio_channel *channels;
	struct iio_dev *indio_dev;
};

static struct iio_cb_buffer *buffer_to_cb_buffer(struct iio_buffer *buffer)
{
	return container_of(buffer, struct iio_cb_buffer, buffer);
}

static int iio_buffer_cb_store_to(struct iio_buffer *buffer, const void *data)
{
	struct iio_cb_buffer *cb_buff = buffer_to_cb_buffer(buffer);
	return cb_buff->cb(data, cb_buff->private);
}

static void iio_buffer_cb_release(struct iio_buffer *buffer)
{
	struct iio_cb_buffer *cb_buff = buffer_to_cb_buffer(buffer);
	kfree(cb_buff->buffer.scan_mask);
	kfree(cb_buff);
}

static const struct iio_buffer_access_funcs iio_cb_access = {
	.store_to = &iio_buffer_cb_store_to,
	.release = &iio_buffer_cb_release,

	.modes = INDIO_BUFFER_SOFTWARE | INDIO_BUFFER_TRIGGERED,
};

struct iio_cb_buffer *iio_channel_get_all_cb(struct device *dev,
					     int (*cb)(const void *data,
						       void *private),
					     void *private)
{
	int ret;
	struct iio_cb_buffer *cb_buff;
	struct iio_channel *chan;

	cb_buff = kzalloc(sizeof(*cb_buff), GFP_KERNEL);
	if (cb_buff == NULL)
		return ERR_PTR(-ENOMEM);

	iio_buffer_init(&cb_buff->buffer);

	cb_buff->private = private;
	cb_buff->cb = cb;
	cb_buff->buffer.access = &iio_cb_access;
	INIT_LIST_HEAD(&cb_buff->buffer.demux_list);

	cb_buff->channels = iio_channel_get_all(dev);
	if (IS_ERR(cb_buff->channels)) {
		ret = PTR_ERR(cb_buff->channels);
		goto error_free_cb_buff;
	}

	cb_buff->indio_dev = cb_buff->channels[0].indio_dev;
	cb_buff->buffer.scan_mask
		= kcalloc(BITS_TO_LONGS(cb_buff->indio_dev->masklength),
			  sizeof(long), GFP_KERNEL);
	if (cb_buff->buffer.scan_mask == NULL) {
		ret = -ENOMEM;
		goto error_release_channels;
	}
	chan = &cb_buff->channels[0];
	while (chan->indio_dev) {
		if (chan->indio_dev != cb_buff->indio_dev) {
			ret = -EINVAL;
			goto error_free_scan_mask;
		}
		set_bit(chan->channel->scan_index,
			cb_buff->buffer.scan_mask);
		chan++;
	}

	return cb_buff;

error_free_scan_mask:
	kfree(cb_buff->buffer.scan_mask);
error_release_channels:
	iio_channel_release_all(cb_buff->channels);
error_free_cb_buff:
	kfree(cb_buff);
	return ERR_PTR(ret);
}
EXPORT_SYMBOL_GPL(iio_channel_get_all_cb);

int iio_channel_cb_set_buffer_watermark(struct iio_cb_buffer *cb_buff,
					size_t watermark)
{
	if (!watermark)
		return -EINVAL;
	cb_buff->buffer.watermark = watermark;

	return 0;
}
EXPORT_SYMBOL_GPL(iio_channel_cb_set_buffer_watermark);

int iio_channel_start_all_cb(struct iio_cb_buffer *cb_buff)
{
	return iio_update_buffers(cb_buff->indio_dev, &cb_buff->buffer,
				  NULL);
}
EXPORT_SYMBOL_GPL(iio_channel_start_all_cb);

void iio_channel_stop_all_cb(struct iio_cb_buffer *cb_buff)
{
	iio_update_buffers(cb_buff->indio_dev, NULL, &cb_buff->buffer);
}
EXPORT_SYMBOL_GPL(iio_channel_stop_all_cb);

void iio_channel_release_all_cb(struct iio_cb_buffer *cb_buff)
{
	iio_channel_release_all(cb_buff->channels);
	iio_buffer_put(&cb_buff->buffer);
}
EXPORT_SYMBOL_GPL(iio_channel_release_all_cb);

struct iio_channel
*iio_channel_cb_get_channels(const struct iio_cb_buffer *cb_buffer)
{
	return cb_buffer->channels;
}
EXPORT_SYMBOL_GPL(iio_channel_cb_get_channels);

struct iio_dev
*iio_channel_cb_get_iio_dev(const struct iio_cb_buffer *cb_buffer)
{
	return cb_buffer->indio_dev;
}
EXPORT_SYMBOL_GPL(iio_channel_cb_get_iio_dev);

MODULE_AUTHOR("Jonathan Cameron <jic23@kernel.org>");
MODULE_DESCRIPTION("Industrial I/O callback buffer");
MODULE_LICENSE("GPL");
Commit	Line	Data
050ee2f1 LPC	1	/* The industrial I/O callback buffer
	2	*
	3	* This program is free software; you can redistribute it and/or modify it
	4	* under the terms of the GNU General Public License version 2 as published by
	5	* the Free Software Foundation.
	6	*/
	7
92d1079b	8	#include <linux/kernel.h>
050ee2f1	9	#include <linux/module.h>
92d1079b JC	10	#include <linux/slab.h>
	11	#include <linux/err.h>
	12	#include <linux/export.h>
19a5a8a5	13	#include <linux/iio/iio.h>
33dd94cb	14	#include <linux/iio/buffer_impl.h>
92d1079b JC	15	#include <linux/iio/consumer.h>
	16
	17	struct iio_cb_buffer {
	18	struct iio_buffer buffer;
5d65d920	19	int (cb)(const void data, void *private);
92d1079b JC	20	void *private;
92d1079b JC	21	struct iio_channel *channels;
adca058b	22	struct iio_dev *indio_dev;
92d1079b JC	23	};
92d1079b JC	24
9e69c935	25	static struct iio_cb_buffer buffer_to_cb_buffer(struct iio_buffer buffer)
92d1079b	26	{
9e69c935 LPC	27	return container_of(buffer, struct iio_cb_buffer, buffer);
9e69c935 LPC	28	}
92d1079b	29
9e69c935 LPC	30	static int iio_buffer_cb_store_to(struct iio_buffer buffer, const void data)
	31	{
	32	struct iio_cb_buffer *cb_buff = buffer_to_cb_buffer(buffer);
92d1079b JC	33	return cb_buff->cb(data, cb_buff->private);
	34	}
	35
9e69c935 LPC	36	static void iio_buffer_cb_release(struct iio_buffer *buffer)
	37	{
	38	struct iio_cb_buffer *cb_buff = buffer_to_cb_buffer(buffer);
	39	kfree(cb_buff->buffer.scan_mask);
	40	kfree(cb_buff);
	41	}
	42
c7a22c36	43	static const struct iio_buffer_access_funcs iio_cb_access = {
92d1079b	44	.store_to = &iio_buffer_cb_store_to,
9e69c935	45	.release = &iio_buffer_cb_release,
225d59ad LPC	46
225d59ad LPC	47	.modes = INDIO_BUFFER_SOFTWARE \| INDIO_BUFFER_TRIGGERED,
92d1079b JC	48	};
92d1079b JC	49
ca7d98db	50	struct iio_cb_buffer iio_channel_get_all_cb(struct device dev,
5d65d920	51	int (cb)(const void data,
92d1079b JC	52	void *private),
	53	void *private)
	54	{
	55	int ret;
	56	struct iio_cb_buffer *cb_buff;
92d1079b JC	57	struct iio_channel *chan;
	58
	59	cb_buff = kzalloc(sizeof(*cb_buff), GFP_KERNEL);
92825ff9 HK	60	if (cb_buff == NULL)
92825ff9 HK	61	return ERR_PTR(-ENOMEM);
92d1079b	62
bda2f8fc LPC	63	iio_buffer_init(&cb_buff->buffer);
bda2f8fc LPC	64
92d1079b JC	65	cb_buff->private = private;
	66	cb_buff->cb = cb;
	67	cb_buff->buffer.access = &iio_cb_access;
	68	INIT_LIST_HEAD(&cb_buff->buffer.demux_list);
	69
ca7d98db	70	cb_buff->channels = iio_channel_get_all(dev);
92d1079b JC	71	if (IS_ERR(cb_buff->channels)) {
	72	ret = PTR_ERR(cb_buff->channels);
	73	goto error_free_cb_buff;
	74	}
	75
adca058b	76	cb_buff->indio_dev = cb_buff->channels[0].indio_dev;
92d1079b	77	cb_buff->buffer.scan_mask
adca058b MR	78	= kcalloc(BITS_TO_LONGS(cb_buff->indio_dev->masklength),
adca058b MR	79	sizeof(long), GFP_KERNEL);
92d1079b JC	80	if (cb_buff->buffer.scan_mask == NULL) {
	81	ret = -ENOMEM;
	82	goto error_release_channels;
	83	}
	84	chan = &cb_buff->channels[0];
	85	while (chan->indio_dev) {
adca058b	86	if (chan->indio_dev != cb_buff->indio_dev) {
92d1079b	87	ret = -EINVAL;
702df9f1	88	goto error_free_scan_mask;
92d1079b JC	89	}
	90	set_bit(chan->channel->scan_index,
	91	cb_buff->buffer.scan_mask);
	92	chan++;
	93	}
	94
	95	return cb_buff;
	96
702df9f1 JC	97	error_free_scan_mask:
702df9f1 JC	98	kfree(cb_buff->buffer.scan_mask);
92d1079b JC	99	error_release_channels:
	100	iio_channel_release_all(cb_buff->channels);
	101	error_free_cb_buff:
	102	kfree(cb_buff);
92d1079b JC	103	return ERR_PTR(ret);
	104	}
	105	EXPORT_SYMBOL_GPL(iio_channel_get_all_cb);
16cbca06 AP	106
	107	int iio_channel_cb_set_buffer_watermark(struct iio_cb_buffer *cb_buff,
	108	size_t watermark)
	109	{
	110	if (!watermark)
	111	return -EINVAL;
	112	cb_buff->buffer.watermark = watermark;
	113
	114	return 0;
	115	}
	116	EXPORT_SYMBOL_GPL(iio_channel_cb_set_buffer_watermark);
92d1079b JC	117
	118	int iio_channel_start_all_cb(struct iio_cb_buffer *cb_buff)
	119	{
adca058b	120	return iio_update_buffers(cb_buff->indio_dev, &cb_buff->buffer,
92d1079b JC	121	NULL);
	122	}
	123	EXPORT_SYMBOL_GPL(iio_channel_start_all_cb);
	124
	125	void iio_channel_stop_all_cb(struct iio_cb_buffer *cb_buff)
	126	{
adca058b	127	iio_update_buffers(cb_buff->indio_dev, NULL, &cb_buff->buffer);
92d1079b JC	128	}
	129	EXPORT_SYMBOL_GPL(iio_channel_stop_all_cb);
	130
	131	void iio_channel_release_all_cb(struct iio_cb_buffer *cb_buff)
	132	{
	133	iio_channel_release_all(cb_buff->channels);
9e69c935	134	iio_buffer_put(&cb_buff->buffer);
92d1079b JC	135	}
	136	EXPORT_SYMBOL_GPL(iio_channel_release_all_cb);
	137
	138	struct iio_channel
	139	iio_channel_cb_get_channels(const struct iio_cb_buffer cb_buffer)
	140	{
	141	return cb_buffer->channels;
	142	}
	143	EXPORT_SYMBOL_GPL(iio_channel_cb_get_channels);
050ee2f1	144
adca058b MR	145	struct iio_dev
	146	iio_channel_cb_get_iio_dev(const struct iio_cb_buffer cb_buffer)
	147	{
	148	return cb_buffer->indio_dev;
	149	}
	150	EXPORT_SYMBOL_GPL(iio_channel_cb_get_iio_dev);
	151
050ee2f1 LPC	152	MODULE_AUTHOR("Jonathan Cameron <jic23@kernel.org>");
	153	MODULE_DESCRIPTION("Industrial I/O callback buffer");
	154	MODULE_LICENSE("GPL");