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 
8 #include <linux/kernel.h>
9 #include <linux/module.h>
10 #include <linux/slab.h>
11 #include <linux/err.h>
12 #include <linux/export.h>
13 #include <linux/iio/buffer.h>
14 #include <linux/iio/consumer.h>
15 
16 struct iio_cb_buffer {
17 	struct iio_buffer buffer;
18 	int (*cb)(const void *data, void *private);
19 	void *private;
20 	struct iio_channel *channels;
21 };
22 
23 static struct iio_cb_buffer *buffer_to_cb_buffer(struct iio_buffer *buffer)
24 {
25 	return container_of(buffer, struct iio_cb_buffer, buffer);
26 }
27 
28 static int iio_buffer_cb_store_to(struct iio_buffer *buffer, const void *data)
29 {
30 	struct iio_cb_buffer *cb_buff = buffer_to_cb_buffer(buffer);
31 	return cb_buff->cb(data, cb_buff->private);
32 }
33 
34 static void iio_buffer_cb_release(struct iio_buffer *buffer)
35 {
36 	struct iio_cb_buffer *cb_buff = buffer_to_cb_buffer(buffer);
37 	kfree(cb_buff->buffer.scan_mask);
38 	kfree(cb_buff);
39 }
40 
41 static const struct iio_buffer_access_funcs iio_cb_access = {
42 	.store_to = &iio_buffer_cb_store_to,
43 	.release = &iio_buffer_cb_release,
44 
45 	.modes = INDIO_BUFFER_SOFTWARE | INDIO_BUFFER_TRIGGERED,
46 };
47 
48 struct iio_cb_buffer *iio_channel_get_all_cb(struct device *dev,
49 					     int (*cb)(const void *data,
50 						       void *private),
51 					     void *private)
52 {
53 	int ret;
54 	struct iio_cb_buffer *cb_buff;
55 	struct iio_dev *indio_dev;
56 	struct iio_channel *chan;
57 
58 	cb_buff = kzalloc(sizeof(*cb_buff), GFP_KERNEL);
59 	if (cb_buff == NULL)
60 		return ERR_PTR(-ENOMEM);
61 
62 	iio_buffer_init(&cb_buff->buffer);
63 
64 	cb_buff->private = private;
65 	cb_buff->cb = cb;
66 	cb_buff->buffer.access = &iio_cb_access;
67 	INIT_LIST_HEAD(&cb_buff->buffer.demux_list);
68 
69 	cb_buff->channels = iio_channel_get_all(dev);
70 	if (IS_ERR(cb_buff->channels)) {
71 		ret = PTR_ERR(cb_buff->channels);
72 		goto error_free_cb_buff;
73 	}
74 
75 	indio_dev = cb_buff->channels[0].indio_dev;
76 	cb_buff->buffer.scan_mask
77 		= kcalloc(BITS_TO_LONGS(indio_dev->masklength), sizeof(long),
78 			  GFP_KERNEL);
79 	if (cb_buff->buffer.scan_mask == NULL) {
80 		ret = -ENOMEM;
81 		goto error_release_channels;
82 	}
83 	chan = &cb_buff->channels[0];
84 	while (chan->indio_dev) {
85 		if (chan->indio_dev != indio_dev) {
86 			ret = -EINVAL;
87 			goto error_free_scan_mask;
88 		}
89 		set_bit(chan->channel->scan_index,
90 			cb_buff->buffer.scan_mask);
91 		chan++;
92 	}
93 
94 	return cb_buff;
95 
96 error_free_scan_mask:
97 	kfree(cb_buff->buffer.scan_mask);
98 error_release_channels:
99 	iio_channel_release_all(cb_buff->channels);
100 error_free_cb_buff:
101 	kfree(cb_buff);
102 	return ERR_PTR(ret);
103 }
104 EXPORT_SYMBOL_GPL(iio_channel_get_all_cb);
105 
106 int iio_channel_start_all_cb(struct iio_cb_buffer *cb_buff)
107 {
108 	return iio_update_buffers(cb_buff->channels[0].indio_dev,
109 				  &cb_buff->buffer,
110 				  NULL);
111 }
112 EXPORT_SYMBOL_GPL(iio_channel_start_all_cb);
113 
114 void iio_channel_stop_all_cb(struct iio_cb_buffer *cb_buff)
115 {
116 	iio_update_buffers(cb_buff->channels[0].indio_dev,
117 			   NULL,
118 			   &cb_buff->buffer);
119 }
120 EXPORT_SYMBOL_GPL(iio_channel_stop_all_cb);
121 
122 void iio_channel_release_all_cb(struct iio_cb_buffer *cb_buff)
123 {
124 	iio_channel_release_all(cb_buff->channels);
125 	iio_buffer_put(&cb_buff->buffer);
126 }
127 EXPORT_SYMBOL_GPL(iio_channel_release_all_cb);
128 
129 struct iio_channel
130 *iio_channel_cb_get_channels(const struct iio_cb_buffer *cb_buffer)
131 {
132 	return cb_buffer->channels;
133 }
134 EXPORT_SYMBOL_GPL(iio_channel_cb_get_channels);
135 
136 MODULE_AUTHOR("Jonathan Cameron <jic23@kernel.org>");
137 MODULE_DESCRIPTION("Industrial I/O callback buffer");
138 MODULE_LICENSE("GPL");
139