drivers/leds/trigger/ledtrig-oneshot.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * One-shot LED Trigger
   4  *
   5  * Copyright 2012, Fabio Baltieri <[email protected]>
   6  *
   7  * Based on ledtrig-timer.c by Richard Purdie <[email protected]>
   8  */
   9
  10 #include <linux/module.h>
  11 #include <linux/kernel.h>
  12 #include <linux/init.h>
  13 #include <linux/device.h>
  14 #include <linux/ctype.h>
  15 #include <linux/slab.h>
  16 #include <linux/leds.h>
  17 #include "../leds.h"
  18
  19 #define DEFAULT_DELAY 100
  20
  21 struct oneshot_trig_data {
  22         unsigned int invert;
  23 };
  24
  25 static ssize_t led_shot(struct device *dev,
  26                 struct device_attribute *attr, const char *buf, size_t size)
  27 {
  28         struct led_classdev *led_cdev = led_trigger_get_led(dev);
  29         struct oneshot_trig_data *oneshot_data = led_trigger_get_drvdata(dev);
  30
  31         led_blink_set_oneshot(led_cdev,
  32                         &led_cdev->blink_delay_on, &led_cdev->blink_delay_off,
  33                         oneshot_data->invert);
  34
  35         /* content is ignored */
  36         return size;
  37 }
  38 static ssize_t led_invert_show(struct device *dev,
  39                 struct device_attribute *attr, char *buf)
  40 {
  41         struct oneshot_trig_data *oneshot_data = led_trigger_get_drvdata(dev);
  42
  43         return sprintf(buf, "%u\n", oneshot_data->invert);
  44 }
  45
  46 static ssize_t led_invert_store(struct device *dev,
  47                 struct device_attribute *attr, const char *buf, size_t size)
  48 {
  49         struct led_classdev *led_cdev = led_trigger_get_led(dev);
  50         struct oneshot_trig_data *oneshot_data = led_trigger_get_drvdata(dev);
  51         unsigned long state;
  52         int ret;
  53
  54         ret = kstrtoul(buf, 0, &state);
  55         if (ret)
  56                 return ret;
  57
  58         oneshot_data->invert = !!state;
  59
  60         if (oneshot_data->invert)
  61                 led_set_brightness_nosleep(led_cdev, LED_FULL);
  62         else
  63                 led_set_brightness_nosleep(led_cdev, LED_OFF);
  64
  65         return size;
  66 }
  67
  68 static ssize_t led_delay_on_show(struct device *dev,
  69                 struct device_attribute *attr, char *buf)
  70 {
  71         struct led_classdev *led_cdev = led_trigger_get_led(dev);
  72
  73         return sprintf(buf, "%lu\n", led_cdev->blink_delay_on);
  74 }
  75
  76 static ssize_t led_delay_on_store(struct device *dev,
  77                 struct device_attribute *attr, const char *buf, size_t size)
  78 {
  79         struct led_classdev *led_cdev = led_trigger_get_led(dev);
  80         unsigned long state;
  81         int ret;
  82
  83         ret = kstrtoul(buf, 0, &state);
  84         if (ret)
  85                 return ret;
  86
  87         led_cdev->blink_delay_on = state;
  88
  89         return size;
  90 }
  91
  92 static ssize_t led_delay_off_show(struct device *dev,
  93                 struct device_attribute *attr, char *buf)
  94 {
  95         struct led_classdev *led_cdev = led_trigger_get_led(dev);
  96
  97         return sprintf(buf, "%lu\n", led_cdev->blink_delay_off);
  98 }
  99
 100 static ssize_t led_delay_off_store(struct device *dev,
 101                 struct device_attribute *attr, const char *buf, size_t size)
 102 {
 103         struct led_classdev *led_cdev = led_trigger_get_led(dev);
 104         unsigned long state;
 105         int ret;
 106
 107         ret = kstrtoul(buf, 0, &state);
 108         if (ret)
 109                 return ret;
 110
 111         led_cdev->blink_delay_off = state;
 112
 113         return size;
 114 }
 115
 116 static DEVICE_ATTR(delay_on, 0644, led_delay_on_show, led_delay_on_store);
 117 static DEVICE_ATTR(delay_off, 0644, led_delay_off_show, led_delay_off_store);
 118 static DEVICE_ATTR(invert, 0644, led_invert_show, led_invert_store);
 119 static DEVICE_ATTR(shot, 0200, NULL, led_shot);
 120
 121 static struct attribute *oneshot_trig_attrs[] = {
 122         &dev_attr_delay_on.attr,
 123         &dev_attr_delay_off.attr,
 124         &dev_attr_invert.attr,
 125         &dev_attr_shot.attr,
 126         NULL
 127 };
 128 ATTRIBUTE_GROUPS(oneshot_trig);
 129
 130 static void pattern_init(struct led_classdev *led_cdev)
 131 {
 132         u32 *pattern;
 133         unsigned int size = 0;
 134
 135         pattern = led_get_default_pattern(led_cdev, &size);
 136         if (!pattern)
 137                 goto out_default;
 138
 139         if (size != 2) {
 140                 dev_warn(led_cdev->dev,
 141                          "Expected 2 but got %u values for delays pattern\n",
 142                          size);
 143                 goto out_default;
 144         }
 145
 146         led_cdev->blink_delay_on = pattern[0];
 147         led_cdev->blink_delay_off = pattern[1];
 148         kfree(pattern);
 149
 150         return;
 151
 152 out_default:
 153         kfree(pattern);
 154         led_cdev->blink_delay_on = DEFAULT_DELAY;
 155         led_cdev->blink_delay_off = DEFAULT_DELAY;
 156 }
 157
 158 static int oneshot_trig_activate(struct led_classdev *led_cdev)
 159 {
 160         struct oneshot_trig_data *oneshot_data;
 161
 162         oneshot_data = kzalloc(sizeof(*oneshot_data), GFP_KERNEL);
 163         if (!oneshot_data)
 164                 return -ENOMEM;
 165
 166         led_set_trigger_data(led_cdev, oneshot_data);
 167
 168         if (led_cdev->flags & LED_INIT_DEFAULT_TRIGGER) {
 169                 pattern_init(led_cdev);
 170                 /*
 171                  * Mark as initialized even on pattern_init() error because
 172                  * any consecutive call to it would produce the same error.
 173                  */
 174                 led_cdev->flags &= ~LED_INIT_DEFAULT_TRIGGER;
 175         }
 176
 177         return 0;
 178 }
 179
 180 static void oneshot_trig_deactivate(struct led_classdev *led_cdev)
 181 {
 182         struct oneshot_trig_data *oneshot_data = led_get_trigger_data(led_cdev);
 183
 184         kfree(oneshot_data);
 185
 186         /* Stop blinking */
 187         led_set_brightness(led_cdev, LED_OFF);
 188 }
 189
 190 static struct led_trigger oneshot_led_trigger = {
 191         .name     = "oneshot",
 192         .activate = oneshot_trig_activate,
 193         .deactivate = oneshot_trig_deactivate,
 194         .groups = oneshot_trig_groups,
 195 };
 196 module_led_trigger(oneshot_led_trigger);
 197
 198 MODULE_AUTHOR("Fabio Baltieri <[email protected]>");
 199 MODULE_DESCRIPTION("One-shot LED trigger");
 200 MODULE_LICENSE("GPL v2");