1 /*
2  * LED Kernel Transient Trigger
3  *
4  * Copyright (C) 2012 Shuah Khan <shuahkhan@gmail.com>
5  *
6  * Based on Richard Purdie's ledtrig-timer.c and Atsushi Nemoto's
7  * ledtrig-heartbeat.c
8  * Design and use-case input from Jonas Bonn <jonas@southpole.se> and
9  * Neil Brown <neilb@suse.de>
10  *
11  * This program is free software; you can redistribute it and/or modify
12  * it under the terms of the GNU General Public License version 2 as
13  * published by the Free Software Foundation.
14  *
15  */
16 /*
17  * Transient trigger allows one shot timer activation. Please refer to
18  * Documentation/leds/ledtrig-transient.txt for details
19 */
20 
21 #include <linux/module.h>
22 #include <linux/kernel.h>
23 #include <linux/init.h>
24 #include <linux/device.h>
25 #include <linux/slab.h>
26 #include <linux/timer.h>
27 #include <linux/leds.h>
28 #include "../leds.h"
29 
30 struct transient_trig_data {
31 	int activate;
32 	int state;
33 	int restore_state;
34 	unsigned long duration;
35 	struct timer_list timer;
36 };
37 
38 static void transient_timer_function(unsigned long data)
39 {
40 	struct led_classdev *led_cdev = (struct led_classdev *) data;
41 	struct transient_trig_data *transient_data = led_cdev->trigger_data;
42 
43 	transient_data->activate = 0;
44 	led_set_brightness_nosleep(led_cdev, transient_data->restore_state);
45 }
46 
47 static ssize_t transient_activate_show(struct device *dev,
48 		struct device_attribute *attr, char *buf)
49 {
50 	struct led_classdev *led_cdev = dev_get_drvdata(dev);
51 	struct transient_trig_data *transient_data = led_cdev->trigger_data;
52 
53 	return sprintf(buf, "%d\n", transient_data->activate);
54 }
55 
56 static ssize_t transient_activate_store(struct device *dev,
57 		struct device_attribute *attr, const char *buf, size_t size)
58 {
59 	struct led_classdev *led_cdev = dev_get_drvdata(dev);
60 	struct transient_trig_data *transient_data = led_cdev->trigger_data;
61 	unsigned long state;
62 	ssize_t ret;
63 
64 	ret = kstrtoul(buf, 10, &state);
65 	if (ret)
66 		return ret;
67 
68 	if (state != 1 && state != 0)
69 		return -EINVAL;
70 
71 	/* cancel the running timer */
72 	if (state == 0 && transient_data->activate == 1) {
73 		del_timer(&transient_data->timer);
74 		transient_data->activate = state;
75 		led_set_brightness_nosleep(led_cdev,
76 					transient_data->restore_state);
77 		return size;
78 	}
79 
80 	/* start timer if there is no active timer */
81 	if (state == 1 && transient_data->activate == 0 &&
82 	    transient_data->duration != 0) {
83 		transient_data->activate = state;
84 		led_set_brightness_nosleep(led_cdev, transient_data->state);
85 		transient_data->restore_state =
86 		    (transient_data->state == LED_FULL) ? LED_OFF : LED_FULL;
87 		mod_timer(&transient_data->timer,
88 			  jiffies + msecs_to_jiffies(transient_data->duration));
89 	}
90 
91 	/* state == 0 && transient_data->activate == 0
92 		timer is not active - just return */
93 	/* state == 1 && transient_data->activate == 1
94 		timer is already active - just return */
95 
96 	return size;
97 }
98 
99 static ssize_t transient_duration_show(struct device *dev,
100 		struct device_attribute *attr, char *buf)
101 {
102 	struct led_classdev *led_cdev = dev_get_drvdata(dev);
103 	struct transient_trig_data *transient_data = led_cdev->trigger_data;
104 
105 	return sprintf(buf, "%lu\n", transient_data->duration);
106 }
107 
108 static ssize_t transient_duration_store(struct device *dev,
109 		struct device_attribute *attr, const char *buf, size_t size)
110 {
111 	struct led_classdev *led_cdev = dev_get_drvdata(dev);
112 	struct transient_trig_data *transient_data = led_cdev->trigger_data;
113 	unsigned long state;
114 	ssize_t ret;
115 
116 	ret = kstrtoul(buf, 10, &state);
117 	if (ret)
118 		return ret;
119 
120 	transient_data->duration = state;
121 	return size;
122 }
123 
124 static ssize_t transient_state_show(struct device *dev,
125 		struct device_attribute *attr, char *buf)
126 {
127 	struct led_classdev *led_cdev = dev_get_drvdata(dev);
128 	struct transient_trig_data *transient_data = led_cdev->trigger_data;
129 	int state;
130 
131 	state = (transient_data->state == LED_FULL) ? 1 : 0;
132 	return sprintf(buf, "%d\n", state);
133 }
134 
135 static ssize_t transient_state_store(struct device *dev,
136 		struct device_attribute *attr, const char *buf, size_t size)
137 {
138 	struct led_classdev *led_cdev = dev_get_drvdata(dev);
139 	struct transient_trig_data *transient_data = led_cdev->trigger_data;
140 	unsigned long state;
141 	ssize_t ret;
142 
143 	ret = kstrtoul(buf, 10, &state);
144 	if (ret)
145 		return ret;
146 
147 	if (state != 1 && state != 0)
148 		return -EINVAL;
149 
150 	transient_data->state = (state == 1) ? LED_FULL : LED_OFF;
151 	return size;
152 }
153 
154 static DEVICE_ATTR(activate, 0644, transient_activate_show,
155 		   transient_activate_store);
156 static DEVICE_ATTR(duration, 0644, transient_duration_show,
157 		   transient_duration_store);
158 static DEVICE_ATTR(state, 0644, transient_state_show, transient_state_store);
159 
160 static void transient_trig_activate(struct led_classdev *led_cdev)
161 {
162 	int rc;
163 	struct transient_trig_data *tdata;
164 
165 	tdata = kzalloc(sizeof(struct transient_trig_data), GFP_KERNEL);
166 	if (!tdata) {
167 		dev_err(led_cdev->dev,
168 			"unable to allocate transient trigger\n");
169 		return;
170 	}
171 	led_cdev->trigger_data = tdata;
172 
173 	rc = device_create_file(led_cdev->dev, &dev_attr_activate);
174 	if (rc)
175 		goto err_out;
176 
177 	rc = device_create_file(led_cdev->dev, &dev_attr_duration);
178 	if (rc)
179 		goto err_out_duration;
180 
181 	rc = device_create_file(led_cdev->dev, &dev_attr_state);
182 	if (rc)
183 		goto err_out_state;
184 
185 	setup_timer(&tdata->timer, transient_timer_function,
186 		    (unsigned long) led_cdev);
187 	led_cdev->activated = true;
188 
189 	return;
190 
191 err_out_state:
192 	device_remove_file(led_cdev->dev, &dev_attr_duration);
193 err_out_duration:
194 	device_remove_file(led_cdev->dev, &dev_attr_activate);
195 err_out:
196 	dev_err(led_cdev->dev, "unable to register transient trigger\n");
197 	led_cdev->trigger_data = NULL;
198 	kfree(tdata);
199 }
200 
201 static void transient_trig_deactivate(struct led_classdev *led_cdev)
202 {
203 	struct transient_trig_data *transient_data = led_cdev->trigger_data;
204 
205 	if (led_cdev->activated) {
206 		del_timer_sync(&transient_data->timer);
207 		led_set_brightness_nosleep(led_cdev,
208 					transient_data->restore_state);
209 		device_remove_file(led_cdev->dev, &dev_attr_activate);
210 		device_remove_file(led_cdev->dev, &dev_attr_duration);
211 		device_remove_file(led_cdev->dev, &dev_attr_state);
212 		led_cdev->trigger_data = NULL;
213 		led_cdev->activated = false;
214 		kfree(transient_data);
215 	}
216 }
217 
218 static struct led_trigger transient_trigger = {
219 	.name     = "transient",
220 	.activate = transient_trig_activate,
221 	.deactivate = transient_trig_deactivate,
222 };
223 
224 static int __init transient_trig_init(void)
225 {
226 	return led_trigger_register(&transient_trigger);
227 }
228 
229 static void __exit transient_trig_exit(void)
230 {
231 	led_trigger_unregister(&transient_trigger);
232 }
233 
234 module_init(transient_trig_init);
235 module_exit(transient_trig_exit);
236 
237 MODULE_AUTHOR("Shuah Khan <shuahkhan@gmail.com>");
238 MODULE_DESCRIPTION("Transient LED trigger");
239 MODULE_LICENSE("GPL");
240