drivers/leds/trigger/ledtrig-heartbeat.c

   1 /*
   2  * LED Heartbeat Trigger
   3  *
   4  * Copyright (C) 2006 Atsushi Nemoto <anemo@mba.ocn.ne.jp>
   5  *
   6  * Based on Richard Purdie's ledtrig-timer.c and some arch's
   7  * CONFIG_HEARTBEAT code.
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License version 2 as
  11  * published by the Free Software Foundation.
  12  *
  13  */
  14 #include <linux/module.h>
  15 #include <linux/kernel.h>
  16 #include <linux/init.h>
  17 #include <linux/slab.h>
  18 #include <linux/timer.h>
  19 #include <linux/sched.h>
  20 #include <linux/sched/loadavg.h>
  21 #include <linux/leds.h>
  22 #include <linux/reboot.h>
  23 #include "../leds.h"
  24
  25 static int panic_heartbeats;
  26
  27 struct heartbeat_trig_data {
  28         struct led_classdev *led_cdev;
  29         unsigned int phase;
  30         unsigned int period;
  31         struct timer_list timer;
  32         unsigned int invert;
  33 };
  34
  35 static void led_heartbeat_function(struct timer_list *t)
  36 {
  37         struct heartbeat_trig_data *heartbeat_data =
  38                 from_timer(heartbeat_data, t, timer);
  39         struct led_classdev *led_cdev;
  40         unsigned long brightness = LED_OFF;
  41         unsigned long delay = 0;
  42
  43         led_cdev = heartbeat_data->led_cdev;
  44
  45         if (unlikely(panic_heartbeats)) {
  46                 led_set_brightness_nosleep(led_cdev, LED_OFF);
  47                 return;
  48         }
  49
  50         if (test_and_clear_bit(LED_BLINK_BRIGHTNESS_CHANGE, &led_cdev->work_flags))
  51                 led_cdev->blink_brightness = led_cdev->new_blink_brightness;
  52
  53         /* acts like an actual heart beat -- ie thump-thump-pause... */
  54         switch (heartbeat_data->phase) {
  55         case 0:
  56                 /*
  57                  * The hyperbolic function below modifies the
  58                  * heartbeat period length in dependency of the
  59                  * current (1min) load. It goes through the points
  60                  * f(0)=1260, f(1)=860, f(5)=510, f(inf)->300.
  61                  */
  62                 heartbeat_data->period = 300 +
  63                         (6720 << FSHIFT) / (5 * avenrun[0] + (7 << FSHIFT));
  64                 heartbeat_data->period =
  65                         msecs_to_jiffies(heartbeat_data->period);
  66                 delay = msecs_to_jiffies(70);
  67                 heartbeat_data->phase++;
  68                 if (!heartbeat_data->invert)
  69                         brightness = led_cdev->blink_brightness;
  70                 break;
  71         case 1:
  72                 delay = heartbeat_data->period / 4 - msecs_to_jiffies(70);
  73                 heartbeat_data->phase++;
  74                 if (heartbeat_data->invert)
  75                         brightness = led_cdev->blink_brightness;
  76                 break;
  77         case 2:
  78                 delay = msecs_to_jiffies(70);
  79                 heartbeat_data->phase++;
  80                 if (!heartbeat_data->invert)
  81                         brightness = led_cdev->blink_brightness;
  82                 break;
  83         default:
  84                 delay = heartbeat_data->period - heartbeat_data->period / 4 -
  85                         msecs_to_jiffies(70);
  86                 heartbeat_data->phase = 0;
  87                 if (heartbeat_data->invert)
  88                         brightness = led_cdev->blink_brightness;
  89                 break;
  90         }
  91
  92         led_set_brightness_nosleep(led_cdev, brightness);
  93         mod_timer(&heartbeat_data->timer, jiffies + delay);
  94 }
  95
  96 static ssize_t led_invert_show(struct device *dev,
  97                 struct device_attribute *attr, char *buf)
  98 {
  99         struct led_classdev *led_cdev = dev_get_drvdata(dev);
 100         struct heartbeat_trig_data *heartbeat_data = led_cdev->trigger_data;
 101
 102         return sprintf(buf, "%u\n", heartbeat_data->invert);
 103 }
 104
 105 static ssize_t led_invert_store(struct device *dev,
 106                 struct device_attribute *attr, const char *buf, size_t size)
 107 {
 108         struct led_classdev *led_cdev = dev_get_drvdata(dev);
 109         struct heartbeat_trig_data *heartbeat_data = led_cdev->trigger_data;
 110         unsigned long state;
 111         int ret;
 112
 113         ret = kstrtoul(buf, 0, &state);
 114         if (ret)
 115                 return ret;
 116
 117         heartbeat_data->invert = !!state;
 118
 119         return size;
 120 }
 121
 122 static DEVICE_ATTR(invert, 0644, led_invert_show, led_invert_store);
 123
 124 static void heartbeat_trig_activate(struct led_classdev *led_cdev)
 125 {
 126         struct heartbeat_trig_data *heartbeat_data;
 127         int rc;
 128
 129         heartbeat_data = kzalloc(sizeof(*heartbeat_data), GFP_KERNEL);
 130         if (!heartbeat_data)
 131                 return;
 132
 133         led_cdev->trigger_data = heartbeat_data;
 134         heartbeat_data->led_cdev = led_cdev;
 135         rc = device_create_file(led_cdev->dev, &dev_attr_invert);
 136         if (rc) {
 137                 kfree(led_cdev->trigger_data);
 138                 return;
 139         }
 140
 141         timer_setup(&heartbeat_data->timer, led_heartbeat_function, 0);
 142         heartbeat_data->phase = 0;
 143         if (!led_cdev->blink_brightness)
 144                 led_cdev->blink_brightness = led_cdev->max_brightness;
 145         led_heartbeat_function(&heartbeat_data->timer);
 146         set_bit(LED_BLINK_SW, &led_cdev->work_flags);
 147         led_cdev->activated = true;
 148 }
 149
 150 static void heartbeat_trig_deactivate(struct led_classdev *led_cdev)
 151 {
 152         struct heartbeat_trig_data *heartbeat_data = led_cdev->trigger_data;
 153
 154         if (led_cdev->activated) {
 155                 del_timer_sync(&heartbeat_data->timer);
 156                 device_remove_file(led_cdev->dev, &dev_attr_invert);
 157                 kfree(heartbeat_data);
 158                 clear_bit(LED_BLINK_SW, &led_cdev->work_flags);
 159                 led_cdev->activated = false;
 160         }
 161 }
 162
 163 static struct led_trigger heartbeat_led_trigger = {
 164         .name     = "heartbeat",
 165         .activate = heartbeat_trig_activate,
 166         .deactivate = heartbeat_trig_deactivate,
 167 };
 168
 169 static int heartbeat_reboot_notifier(struct notifier_block *nb,
 170                                      unsigned long code, void *unused)
 171 {
 172         led_trigger_unregister(&heartbeat_led_trigger);
 173         return NOTIFY_DONE;
 174 }
 175
 176 static int heartbeat_panic_notifier(struct notifier_block *nb,
 177                                      unsigned long code, void *unused)
 178 {
 179         panic_heartbeats = 1;
 180         return NOTIFY_DONE;
 181 }
 182
 183 static struct notifier_block heartbeat_reboot_nb = {
 184         .notifier_call = heartbeat_reboot_notifier,
 185 };
 186
 187 static struct notifier_block heartbeat_panic_nb = {
 188         .notifier_call = heartbeat_panic_notifier,
 189 };
 190
 191 static int __init heartbeat_trig_init(void)
 192 {
 193         int rc = led_trigger_register(&heartbeat_led_trigger);
 194
 195         if (!rc) {
 196                 atomic_notifier_chain_register(&panic_notifier_list,
 197                                                &heartbeat_panic_nb);
 198                 register_reboot_notifier(&heartbeat_reboot_nb);
 199         }
 200         return rc;
 201 }
 202
 203 static void __exit heartbeat_trig_exit(void)
 204 {
 205         unregister_reboot_notifier(&heartbeat_reboot_nb);
 206         atomic_notifier_chain_unregister(&panic_notifier_list,
 207                                          &heartbeat_panic_nb);
 208         led_trigger_unregister(&heartbeat_led_trigger);
 209 }
 210
 211 module_init(heartbeat_trig_init);
 212 module_exit(heartbeat_trig_exit);
 213
 214 MODULE_AUTHOR("Atsushi Nemoto <anemo@mba.ocn.ne.jp>");
 215 MODULE_DESCRIPTION("Heartbeat LED trigger");
 216 MODULE_LICENSE("GPL");