For the buffer attributes that are present inside the IIO core buffer logic
we can re-route them to expand the attribute into iio_buffer objects.

The rest, will still expand to device_attributes.

Signed-off-by: Alexandru Ardelean <alexandru.ardel...@analog.com>
---
 drivers/iio/industrialio-buffer.c | 113 +++++++++++++++++-------------
 1 file changed, 64 insertions(+), 49 deletions(-)

diff --git a/drivers/iio/industrialio-buffer.c 
b/drivers/iio/industrialio-buffer.c
index 445709ef245c..8b4b7baf21da 100644
--- a/drivers/iio/industrialio-buffer.c
+++ b/drivers/iio/industrialio-buffer.c
@@ -572,22 +572,18 @@ static int iio_buffer_add_channel_sysfs(struct iio_dev 
*indio_dev,
        return ret;
 }
 
-static ssize_t iio_buffer_read_length(struct device *dev,
-                                     struct device_attribute *attr,
+static ssize_t iio_buffer_read_length(struct iio_buffer *buffer,
+                                     struct iio_buf_attr *attr,
                                      char *buf)
 {
-       struct iio_dev *indio_dev = dev_to_iio_dev(dev);
-       struct iio_buffer *buffer = indio_dev->buffer;
-
        return sprintf(buf, "%d\n", buffer->length);
 }
 
-static ssize_t iio_buffer_write_length(struct device *dev,
-                                      struct device_attribute *attr,
+static ssize_t iio_buffer_write_length(struct iio_buffer *buffer,
+                                      struct iio_buf_attr *attr,
                                       const char *buf, size_t len)
 {
-       struct iio_dev *indio_dev = dev_to_iio_dev(dev);
-       struct iio_buffer *buffer = indio_dev->buffer;
+       struct iio_dev *indio_dev = buffer->indio_dev;
        unsigned int val;
        int ret;
 
@@ -615,13 +611,10 @@ static ssize_t iio_buffer_write_length(struct device *dev,
        return ret ? ret : len;
 }
 
-static ssize_t iio_buffer_show_enable(struct device *dev,
-                                     struct device_attribute *attr,
+static ssize_t iio_buffer_show_enable(struct iio_buffer *buffer,
+                                     struct iio_buf_attr *attr,
                                      char *buf)
 {
-       struct iio_dev *indio_dev = dev_to_iio_dev(dev);
-       struct iio_buffer *buffer = indio_dev->buffer;
-
        return sprintf(buf, "%d\n", iio_buffer_is_active(buffer));
 }
 
@@ -1227,15 +1220,14 @@ void iio_disable_all_buffers(struct iio_dev *indio_dev)
        iio_buffer_deactivate_all(indio_dev);
 }
 
-static ssize_t iio_buffer_store_enable(struct device *dev,
-                                      struct device_attribute *attr,
+static ssize_t iio_buffer_store_enable(struct iio_buffer *buffer,
+                                      struct iio_buf_attr *attr,
                                       const char *buf,
                                       size_t len)
 {
        int ret;
        bool requested_state;
-       struct iio_dev *indio_dev = dev_to_iio_dev(dev);
-       struct iio_buffer *buffer = indio_dev->buffer;
+       struct iio_dev *indio_dev = buffer->indio_dev;
        bool inlist;
 
        ret = strtobool(buf, &requested_state);
@@ -1260,23 +1252,19 @@ static ssize_t iio_buffer_store_enable(struct device 
*dev,
        return (ret < 0) ? ret : len;
 }
 
-static ssize_t iio_buffer_show_watermark(struct device *dev,
-                                        struct device_attribute *attr,
+static ssize_t iio_buffer_show_watermark(struct iio_buffer *buffer,
+                                        struct iio_buf_attr *attr,
                                         char *buf)
 {
-       struct iio_dev *indio_dev = dev_to_iio_dev(dev);
-       struct iio_buffer *buffer = indio_dev->buffer;
-
        return sprintf(buf, "%u\n", buffer->watermark);
 }
 
-static ssize_t iio_buffer_store_watermark(struct device *dev,
-                                         struct device_attribute *attr,
+static ssize_t iio_buffer_store_watermark(struct iio_buffer *buffer,
+                                         struct iio_buf_attr *attr,
                                          const char *buf,
                                          size_t len)
 {
-       struct iio_dev *indio_dev = dev_to_iio_dev(dev);
-       struct iio_buffer *buffer = indio_dev->buffer;
+       struct iio_dev *indio_dev = buffer->indio_dev;
        unsigned int val;
        int ret;
 
@@ -1305,36 +1293,51 @@ static ssize_t iio_buffer_store_watermark(struct device 
*dev,
        return ret ? ret : len;
 }
 
-static ssize_t iio_dma_show_data_available(struct device *dev,
-                                               struct device_attribute *attr,
-                                               char *buf)
+static ssize_t iio_dma_show_data_available(struct iio_buffer *buffer,
+                                          struct iio_buf_attr *attr,
+                                          char *buf)
 {
-       struct iio_dev *indio_dev = dev_to_iio_dev(dev);
-       struct iio_buffer *buffer = indio_dev->buffer;
-
        return sprintf(buf, "%zu\n", iio_buffer_data_available(buffer));
 }
 
-static DEVICE_ATTR(length, S_IRUGO | S_IWUSR, iio_buffer_read_length,
-                  iio_buffer_write_length);
-static struct device_attribute dev_attr_length_ro = __ATTR(length,
+#define IIO_BUF_ATTR(_name, _mode, _show, _store)      \
+       struct iio_buf_attr iio_buf_attr_##_name =      \
+               __ATTR(_name, _mode, _show, _store)
+
+static IIO_BUF_ATTR(length, S_IRUGO | S_IWUSR,
+                   iio_buffer_read_length, iio_buffer_write_length);
+static struct iio_buf_attr buf_attr_length_ro = __ATTR(length,
        S_IRUGO, iio_buffer_read_length, NULL);
-static DEVICE_ATTR(enable, S_IRUGO | S_IWUSR,
-                  iio_buffer_show_enable, iio_buffer_store_enable);
-static DEVICE_ATTR(watermark, S_IRUGO | S_IWUSR,
-                  iio_buffer_show_watermark, iio_buffer_store_watermark);
-static struct device_attribute dev_attr_watermark_ro = __ATTR(watermark,
+static IIO_BUF_ATTR(enable, S_IRUGO | S_IWUSR,
+                   iio_buffer_show_enable, iio_buffer_store_enable);
+static IIO_BUF_ATTR(watermark, S_IRUGO | S_IWUSR,
+                   iio_buffer_show_watermark, iio_buffer_store_watermark);
+static struct iio_buf_attr buf_attr_watermark_ro = __ATTR(watermark,
        S_IRUGO, iio_buffer_show_watermark, NULL);
-static DEVICE_ATTR(data_available, S_IRUGO,
-               iio_dma_show_data_available, NULL);
+static IIO_BUF_ATTR(data_available, S_IRUGO,
+                   iio_dma_show_data_available, NULL);
 
 static struct attribute *iio_buffer_attrs[] = {
-       &dev_attr_length.attr,
-       &dev_attr_enable.attr,
-       &dev_attr_watermark.attr,
-       &dev_attr_data_available.attr,
+       &iio_buf_attr_length.attr,
+       &iio_buf_attr_enable.attr,
+       &iio_buf_attr_watermark.attr,
+       &iio_buf_attr_data_available.attr,
 };
 
+static bool iio_buffer_attr_is_core(struct attribute *attr)
+{
+       struct attribute *a;
+       int i;
+
+       for (i = 0; i < ARRAY_SIZE(iio_buffer_attrs); i++) {
+               a = iio_buffer_attrs[i];
+               if (!strcmp(attr->name, a->name))
+                       return true;
+       }
+
+       return false;
+}
+
 #define to_dev_attr(_attr) container_of(_attr, struct device_attribute, attr)
 #define to_iio_buf_attr(_attr) container_of(_attr, struct iio_buf_attr, attr)
 
@@ -1344,6 +1347,12 @@ static ssize_t iio_buffer_dir_attr_show(struct kobject 
*kobj,
 {
        struct iio_buffer *buffer = container_of(kobj, struct iio_buffer, 
buffer_dir);
        struct device_attribute *dattr;
+       struct iio_buf_attr *battr;
+
+       if (iio_buffer_attr_is_core(attr)) {
+               battr = to_iio_buf_attr(attr);
+               return battr->show(buffer, battr, buf);
+       }
 
        dattr = to_dev_attr(attr);
 
@@ -1357,6 +1366,12 @@ static ssize_t iio_buffer_dir_attr_store(struct kobject 
*kobj,
 {
        struct iio_buffer *buffer = container_of(kobj, struct iio_buffer, 
buffer_dir);
        struct device_attribute *dattr;
+       struct iio_buf_attr *battr;
+
+       if (iio_buffer_attr_is_core(attr)) {
+               battr = to_iio_buf_attr(attr);
+               return battr->store(buffer, battr, buf, len);
+       }
 
        dattr = to_dev_attr(attr);
 
@@ -1453,10 +1468,10 @@ static int __iio_buffer_alloc_sysfs_and_mask(struct 
iio_buffer *buffer,
 
        memcpy(attr, iio_buffer_attrs, sizeof(iio_buffer_attrs));
        if (!buffer->access->set_length)
-               attr[0] = &dev_attr_length_ro.attr;
+               attr[0] = &buf_attr_length_ro.attr;
 
        if (buffer->access->flags & INDIO_BUFFER_FLAG_FIXED_WATERMARK)
-               attr[2] = &dev_attr_watermark_ro.attr;
+               attr[2] = &buf_attr_watermark_ro.attr;
 
        if (buffer->attrs)
                memcpy(&attr[ARRAY_SIZE(iio_buffer_attrs)], buffer->attrs,
-- 
2.17.1

Reply via email to