* New class attributes - /sys/class/block/led_trigger_{new,list,del}

* Add init function - blk_ledtrig_init() - to create the attributes
  in sysfs.  Call blk_ledtrig_init() from genhd_device_init() (in
  block/genhd.c).

* New file - block/blk-ledtrig.h

Signed-off-by: Ian Pilcher <[email protected]>
---
 block/blk-ledtrig.c | 147 ++++++++++++++++++++++++++++++++++++++++++++
 block/blk-ledtrig.h |  22 +++++++
 block/genhd.c       |   2 +
 3 files changed, 171 insertions(+)
 create mode 100644 block/blk-ledtrig.h

diff --git a/block/blk-ledtrig.c b/block/blk-ledtrig.c
index c69ea1539336..6392ab4169f9 100644
--- a/block/blk-ledtrig.c
+++ b/block/blk-ledtrig.c
@@ -7,11 +7,15 @@
  */
 
 #include <linux/blk-ledtrig.h>
+#include <linux/ctype.h>
+#include <linux/genhd.h>
 #include <linux/leds.h>
 #include <linux/list.h>
 #include <linux/mutex.h>
 #include <linux/slab.h>
 
+#include "blk-ledtrig.h"
+
 
 /*
  *
@@ -201,3 +205,146 @@ int blk_ledtrig_delete(const char *const name)
        return __blk_ledtrig_delete(name, strlen(name));
 }
 EXPORT_SYMBOL_GPL(blk_ledtrig_delete);
+
+
+/*
+ *
+ *     Class attributes to manage the trigger list
+ *
+ */
+
+static ssize_t blk_ledtrig_attr_store(struct class *, struct class_attribute *,
+                                     const char *, const size_t);
+static ssize_t blk_ledtrig_list_show(struct class *, struct class_attribute *,
+                                    char *);
+
+static struct class_attribute blk_ledtrig_attr_new =
+       __ATTR(led_trigger_new, 0200, 0, blk_ledtrig_attr_store);
+
+static struct class_attribute blk_ledtrig_attr_del =
+       __ATTR(led_trigger_del, 0200, 0, blk_ledtrig_attr_store);
+
+static struct class_attribute blk_ledtrig_attr_list =
+       __ATTR(led_trigger_list, 0444, blk_ledtrig_list_show, 0);
+
+// Returns a pointer to the first non-whitespace character in s (or a pointer
+// to the terminating nul).
+static const char *blk_ledtrig_skip_whitespace(const char *s)
+{
+       while (*s != 0 && isspace(*s))
+               ++s;
+
+       return s;
+}
+
+// Returns a pointer to the first whitespace character in s (or a pointer to
+// the terminating nul), which is effectively a pointer to the position *after*
+// the last character in the non-whitespace token at the beginning of s.  (s is
+// expected to be the result of a previous call to 
blk_ledtrig_skip_whitespace.)
+static const char *blk_ledtrig_find_whitespace(const char *s)
+{
+       while (*s != 0 && !isspace(*s))
+               ++s;
+
+       return s;
+}
+
+static ssize_t blk_ledtrig_attr_store(struct class *const class,
+                                     struct class_attribute *const attr,
+                                     const char *const buf,
+                                     const size_t count)
+{
+       const char *const name = blk_ledtrig_skip_whitespace(buf);
+       const char *const endp = blk_ledtrig_find_whitespace(name);
+       const ptrdiff_t name_len = endp - name;         // always >= 0
+       int ret;
+
+       if (attr == &blk_ledtrig_attr_new)
+               ret = __blk_ledtrig_create(name, name_len);
+       else    // attr == &blk_ledtrig_attr_del
+               ret = __blk_ledtrig_delete(name, name_len);
+
+       if (ret < 0)
+               return ret;
+
+       // Avoid potential "empty name" error by skipping whitespace
+       // to next token or terminating nul
+       return blk_ledtrig_skip_whitespace(endp) - buf;
+}
+
+static ssize_t blk_ledtrig_list_show(struct class *const class,
+                                    struct class_attribute *const attr,
+                                    char *const buf)
+{
+       struct list_head *n;
+       int ret, c = 0;
+
+       ret = mutex_lock_interruptible(&blk_ledtrig_list_mutex);
+       if (unlikely(ret != 0))
+               goto list_exit_return;
+
+       list_for_each(n, &blk_ledtrig_list) {
+
+               struct blk_ledtrig *const t = blk_ledtrig_from_node(n);
+               int refcount;
+
+               ret = mutex_lock_interruptible(&t->refcount_mutex);
+               if (unlikely(ret != 0))
+                       goto list_exit_unlock_list;
+
+               refcount = t->refcount;
+
+               mutex_unlock(&t->refcount_mutex);
+
+               ret = snprintf(buf + c, PAGE_SIZE - c, "%s: %d\n",
+                              t->name, refcount);
+               if (unlikely(ret < 0))
+                       goto list_exit_unlock_list;
+
+               c += ret;
+               if (unlikely(c >= PAGE_SIZE)) {
+                       ret = -EOVERFLOW;
+                       goto list_exit_unlock_list;
+               }
+       }
+
+       ret = c;
+
+list_exit_unlock_list:
+       mutex_unlock(&blk_ledtrig_list_mutex);
+list_exit_return:
+       return ret;
+}
+
+
+/*
+ *
+ *     Initialization - create the class attributes
+ *
+ */
+
+void __init blk_ledtrig_init(void)
+{
+       int ret;
+
+       ret = class_create_file(&block_class, &blk_ledtrig_attr_new);
+       if (unlikely(ret != 0))
+               goto init_error_new;
+
+       ret = class_create_file(&block_class, &blk_ledtrig_attr_del);
+       if (unlikely(ret != 0))
+               goto init_error_del;
+
+       ret = class_create_file(&block_class, &blk_ledtrig_attr_list);
+       if (unlikely(ret != 0))
+               goto init_error_list;
+
+       return;
+
+init_error_list:
+       class_remove_file(&block_class, &blk_ledtrig_attr_del);
+init_error_del:
+       class_remove_file(&block_class, &blk_ledtrig_attr_new);
+init_error_new:
+       pr_err("failed to initialize blkdev LED triggers (%d)\n", ret);
+}
diff --git a/block/blk-ledtrig.h b/block/blk-ledtrig.h
new file mode 100644
index 000000000000..894843249deb
--- /dev/null
+++ b/block/blk-ledtrig.h
@@ -0,0 +1,22 @@
+/* SPDX-License-Identifier: GPL-2.0-only */
+
+/*
+ *     Block device LED triggers
+ *
+ *     Copyright 2021 Ian Pilcher <[email protected]>
+ */
+
+#ifndef _BLOCK_BLK_LEDTRIG_H
+#define _BLOCK_BLK_LEDTRIG_H
+
+#ifdef CONFIG_BLK_LED_TRIGGERS
+
+void blk_ledtrig_init(void);
+
+#else  // CONFIG_BLK_LED_TRIGGERS
+
+static inline void blk_ledtrig_init(void) {}
+
+#endif // CONFIG_BLK_LED_TRIGGERS
+
+#endif // _BLOCK_BLK_LEDTRIG_H
diff --git a/block/genhd.c b/block/genhd.c
index af4d2ab4a633..d0b1d8f743ae 100644
--- a/block/genhd.c
+++ b/block/genhd.c
@@ -26,6 +26,7 @@
 #include <linux/badblocks.h>
 
 #include "blk.h"
+#include "blk-ledtrig.h"
 
 static struct kobject *block_depr;
 
@@ -824,6 +825,7 @@ static int __init genhd_device_init(void)
        if (unlikely(error))
                return error;
        blk_dev_init();
+       blk_ledtrig_init();
 
        register_blkdev(BLOCK_EXT_MAJOR, "blkext");
 
-- 
2.31.1


_______________________________________________
Kernelnewbies mailing list
[email protected]
https://lists.kernelnewbies.org/mailman/listinfo/kernelnewbies

Reply via email to