xref: /freebsd/sys/dev/led/led.c (revision db33c6f3ae9d1231087710068ee4ea5398aacca7)
1 /*-
2  * SPDX-License-Identifier: Beerware
3  *
4  * ----------------------------------------------------------------------------
5  * "THE BEER-WARE LICENSE" (Revision 42):
6  * <phk@FreeBSD.org> wrote this file.  As long as you retain this notice you
7  * can do whatever you want with this stuff. If we meet some day, and you think
8  * this stuff is worth it, you can buy me a beer in return.   Poul-Henning Kamp
9  * ----------------------------------------------------------------------------
10  *
11  */
12 
13 #include <sys/param.h>
14 #include <sys/conf.h>
15 #include <sys/ctype.h>
16 #include <sys/kernel.h>
17 #include <sys/limits.h>
18 #include <sys/lock.h>
19 #include <sys/malloc.h>
20 #include <sys/mutex.h>
21 #include <sys/queue.h>
22 #include <sys/sbuf.h>
23 #include <sys/sx.h>
24 #include <sys/systm.h>
25 #include <sys/uio.h>
26 
27 #include <dev/led/led.h>
28 
29 struct ledsc {
30 	LIST_ENTRY(ledsc)	list;
31 	char			*name;
32 	void			*private;
33 	int			unit;
34 	led_t			*func;
35 	struct cdev *dev;
36 	struct sbuf		*spec;
37 	char			*str;
38 	char			*ptr;
39 	int			count;
40 	time_t			last_second;
41 };
42 
43 static struct unrhdr *led_unit;
44 static struct mtx led_mtx;
45 static struct sx led_sx;
46 static LIST_HEAD(, ledsc) led_list = LIST_HEAD_INITIALIZER(led_list);
47 static struct callout led_ch;
48 static int blinkers = 0;
49 
50 static MALLOC_DEFINE(M_LED, "LED", "LED driver");
51 
52 static void
53 led_timeout(void *p)
54 {
55 	struct ledsc	*sc;
56 
57 	LIST_FOREACH(sc, &led_list, list) {
58 		if (sc->ptr == NULL)
59 			continue;
60 		if (sc->count > 0) {
61 			sc->count--;
62 			continue;
63 		}
64 		if (*sc->ptr == '.') {
65 			sc->ptr = NULL;
66 			blinkers--;
67 			continue;
68 		} else if (*sc->ptr == 'U' || *sc->ptr == 'u') {
69 			if (sc->last_second == time_second)
70 				continue;
71 			sc->last_second = time_second;
72 			sc->func(sc->private, *sc->ptr == 'U');
73 		} else if (*sc->ptr >= 'a' && *sc->ptr <= 'j') {
74 			sc->func(sc->private, 0);
75 			sc->count = (*sc->ptr & 0xf) - 1;
76 		} else if (*sc->ptr >= 'A' && *sc->ptr <= 'J') {
77 			sc->func(sc->private, 1);
78 			sc->count = (*sc->ptr & 0xf) - 1;
79 		}
80 		sc->ptr++;
81 		if (*sc->ptr == '\0')
82 			sc->ptr = sc->str;
83 	}
84 	if (blinkers > 0)
85 		callout_reset(&led_ch, hz / 10, led_timeout, p);
86 }
87 
88 static int
89 led_state(struct ledsc *sc, struct sbuf **sb, int state)
90 {
91 	struct sbuf *sb2 = NULL;
92 
93 	sb2 = sc->spec;
94 	sc->spec = *sb;
95 	if (*sb != NULL) {
96 		sc->str = sbuf_data(*sb);
97 		if (sc->ptr == NULL) {
98 			blinkers++;
99 			callout_reset(&led_ch, hz / 10, led_timeout, NULL);
100 		}
101 		sc->ptr = sc->str;
102 	} else {
103 		sc->str = NULL;
104 		if (sc->ptr != NULL)
105 			blinkers--;
106 		sc->ptr = NULL;
107 		sc->func(sc->private, state);
108 	}
109 	sc->count = 0;
110 	*sb = sb2;
111 	return(0);
112 }
113 
114 static int
115 led_parse(const char *s, struct sbuf **sb, int *state)
116 {
117 	int i, error;
118 
119 	/*
120 	 * Handle "on" and "off" immediately so people can flash really
121 	 * fast from userland if they want to
122 	 */
123 	if (*s == '0' || *s == '1') {
124 		*state = *s & 1;
125 		return (0);
126 	}
127 
128 	*state = 0;
129 	*sb = sbuf_new_auto();
130 	if (*sb == NULL)
131 		return (ENOMEM);
132 	switch(s[0]) {
133 		/*
134 		 * Flash, default is 100msec/100msec.
135 		 * 'f2' sets 200msec/200msec etc.
136 		 */
137 		case 'f':
138 			if (s[1] >= '1' && s[1] <= '9')
139 				i = s[1] - '1';
140 			else
141 				i = 0;
142 			sbuf_printf(*sb, "%c%c", 'A' + i, 'a' + i);
143 			break;
144 		/*
145 		 * Digits, flashes out numbers.
146 		 * 'd12' becomes -__________-_-______________________________
147 		 */
148 		case 'd':
149 			for(s++; *s; s++) {
150 				if (!isdigit(*s))
151 					continue;
152 				i = *s - '0';
153 				if (i == 0)
154 					i = 10;
155 				for (; i > 1; i--)
156 					sbuf_cat(*sb, "Aa");
157 				sbuf_cat(*sb, "Aj");
158 			}
159 			sbuf_cat(*sb, "jj");
160 			break;
161 		/*
162 		 * String, roll your own.
163 		 * 'a-j' gives "off" for n/10 sec.
164 		 * 'A-J' gives "on" for n/10 sec.
165 		 * no delay before repeat
166 		 * 'sAaAbBa' becomes _-_--__-
167 		 */
168 		case 's':
169 			for(s++; *s; s++) {
170 				if ((*s >= 'a' && *s <= 'j') ||
171 				    (*s >= 'A' && *s <= 'J') ||
172 				    *s == 'U' || *s <= 'u' ||
173 					*s == '.')
174 					sbuf_bcat(*sb, s, 1);
175 			}
176 			break;
177 		/*
178 		 * Morse.
179 		 * '.' becomes _-
180 		 * '-' becomes _---
181 		 * ' ' becomes __
182 		 * '\n' becomes ____
183 		 * 1sec pause between repeats
184 		 * '... --- ...' -> _-_-_-___---_---_---___-_-_-__________
185 		 */
186 		case 'm':
187 			for(s++; *s; s++) {
188 				if (*s == '.')
189 					sbuf_cat(*sb, "aA");
190 				else if (*s == '-')
191 					sbuf_cat(*sb, "aC");
192 				else if (*s == ' ')
193 					sbuf_cat(*sb, "b");
194 				else if (*s == '\n')
195 					sbuf_cat(*sb, "d");
196 			}
197 			sbuf_cat(*sb, "j");
198 			break;
199 		default:
200 			sbuf_delete(*sb);
201 			return (EINVAL);
202 	}
203 	error = sbuf_finish(*sb);
204 	if (error != 0 || sbuf_len(*sb) == 0) {
205 		*sb = NULL;
206 		return (error);
207 	}
208 	return (0);
209 }
210 
211 static int
212 led_write(struct cdev *dev, struct uio *uio, int ioflag)
213 {
214 	struct ledsc	*sc;
215 	char *s;
216 	struct sbuf *sb = NULL;
217 	int error, state = 0;
218 
219 	if (uio->uio_resid > 512)
220 		return (EINVAL);
221 	s = malloc(uio->uio_resid + 1, M_DEVBUF, M_WAITOK);
222 	s[uio->uio_resid] = '\0';
223 	error = uiomove(s, uio->uio_resid, uio);
224 	if (error) {
225 		free(s, M_DEVBUF);
226 		return (error);
227 	}
228 	error = led_parse(s, &sb, &state);
229 	free(s, M_DEVBUF);
230 	if (error)
231 		return (error);
232 	mtx_lock(&led_mtx);
233 	sc = dev->si_drv1;
234 	if (sc != NULL)
235 		error = led_state(sc, &sb, state);
236 	mtx_unlock(&led_mtx);
237 	if (sb != NULL)
238 		sbuf_delete(sb);
239 	return (error);
240 }
241 
242 int
243 led_set(char const *name, char const *cmd)
244 {
245 	struct ledsc	*sc;
246 	struct sbuf *sb = NULL;
247 	int error, state = 0;
248 
249 	error = led_parse(cmd, &sb, &state);
250 	if (error)
251 		return (error);
252 	mtx_lock(&led_mtx);
253 	LIST_FOREACH(sc, &led_list, list) {
254 		if (strcmp(sc->name, name) == 0)
255 			break;
256 	}
257 	if (sc != NULL)
258 		error = led_state(sc, &sb, state);
259 	else
260 		error = ENOENT;
261 	mtx_unlock(&led_mtx);
262 	if (sb != NULL)
263 		sbuf_delete(sb);
264 	return (error);
265 }
266 
267 static struct cdevsw led_cdevsw = {
268 	.d_version =	D_VERSION,
269 	.d_write =	led_write,
270 	.d_name =	"LED",
271 };
272 
273 struct cdev *
274 led_create(led_t *func, void *priv, char const *name)
275 {
276 
277 	return (led_create_state(func, priv, name, 0));
278 }
279 struct cdev *
280 led_create_state(led_t *func, void *priv, char const *name, int state)
281 {
282 	struct ledsc	*sc;
283 
284 	sc = malloc(sizeof *sc, M_LED, M_WAITOK | M_ZERO);
285 
286 	sx_xlock(&led_sx);
287 	sc->name = strdup(name, M_LED);
288 	sc->unit = alloc_unr(led_unit);
289 	sc->private = priv;
290 	sc->func = func;
291 	sc->dev = make_dev(&led_cdevsw, sc->unit,
292 	    UID_ROOT, GID_WHEEL, 0600, "led/%s", name);
293 	sx_xunlock(&led_sx);
294 
295 	mtx_lock(&led_mtx);
296 	sc->dev->si_drv1 = sc;
297 	LIST_INSERT_HEAD(&led_list, sc, list);
298 	if (state != -1)
299 		sc->func(sc->private, state != 0);
300 	mtx_unlock(&led_mtx);
301 
302 	return (sc->dev);
303 }
304 
305 void
306 led_destroy(struct cdev *dev)
307 {
308 	struct ledsc *sc;
309 
310 	mtx_lock(&led_mtx);
311 	sc = dev->si_drv1;
312 	dev->si_drv1 = NULL;
313 	if (sc->ptr != NULL)
314 		blinkers--;
315 	LIST_REMOVE(sc, list);
316 	if (LIST_EMPTY(&led_list))
317 		callout_stop(&led_ch);
318 	mtx_unlock(&led_mtx);
319 
320 	sx_xlock(&led_sx);
321 	free_unr(led_unit, sc->unit);
322 	destroy_dev(dev);
323 	if (sc->spec != NULL)
324 		sbuf_delete(sc->spec);
325 	free(sc->name, M_LED);
326 	free(sc, M_LED);
327 	sx_xunlock(&led_sx);
328 }
329 
330 static void
331 led_drvinit(void *unused)
332 {
333 
334 	led_unit = new_unrhdr(0, INT_MAX, NULL);
335 	mtx_init(&led_mtx, "LED mtx", NULL, MTX_DEF);
336 	sx_init(&led_sx, "LED sx");
337 	callout_init_mtx(&led_ch, &led_mtx, 0);
338 }
339 
340 SYSINIT(leddev, SI_SUB_DRIVERS, SI_ORDER_MIDDLE, led_drvinit, NULL);
341