1121941Sphk/*- 2121941Sphk * ---------------------------------------------------------------------------- 3121941Sphk * "THE BEER-WARE LICENSE" (Revision 42): 4121941Sphk * <phk@FreeBSD.org> wrote this file. As long as you retain this notice you 5121941Sphk * can do whatever you want with this stuff. If we meet some day, and you think 6121941Sphk * this stuff is worth it, you can buy me a beer in return. Poul-Henning Kamp 7121941Sphk * ---------------------------------------------------------------------------- 8121941Sphk * 9121941Sphk */ 10121941Sphk 11121941Sphk#include <sys/cdefs.h> 12121941Sphk__FBSDID("$FreeBSD$"); 13121941Sphk 14121941Sphk#include <sys/param.h> 15121941Sphk#include <sys/conf.h> 16121941Sphk#include <sys/kernel.h> 17121941Sphk#include <sys/systm.h> 18179413Sed#include <sys/limits.h> 19121941Sphk#include <sys/malloc.h> 20121941Sphk#include <sys/ctype.h> 21121941Sphk#include <sys/sbuf.h> 22121941Sphk#include <sys/queue.h> 23121941Sphk#include <dev/led/led.h> 24121941Sphk#include <sys/uio.h> 25140966Sphk#include <sys/sx.h> 26121941Sphk 27121941Sphkstruct ledsc { 28121941Sphk LIST_ENTRY(ledsc) list; 29219951Smav char *name; 30121941Sphk void *private; 31140966Sphk int unit; 32121941Sphk led_t *func; 33130585Sphk struct cdev *dev; 34121941Sphk struct sbuf *spec; 35121941Sphk char *str; 36121941Sphk char *ptr; 37121941Sphk int count; 38140966Sphk time_t last_second; 39121941Sphk}; 40121941Sphk 41140966Sphkstatic struct unrhdr *led_unit; 42121941Sphkstatic struct mtx led_mtx; 43140966Sphkstatic struct sx led_sx; 44201145Santoinestatic LIST_HEAD(, ledsc) led_list = LIST_HEAD_INITIALIZER(led_list); 45140966Sphkstatic struct callout led_ch; 46247008Smavstatic int blinkers = 0; 47121941Sphk 48141616Sphkstatic MALLOC_DEFINE(M_LED, "LED", "LED driver"); 49121941Sphk 50121941Sphkstatic void 51121941Sphkled_timeout(void *p) 52121941Sphk{ 53121941Sphk struct ledsc *sc; 54121941Sphk 55121941Sphk LIST_FOREACH(sc, &led_list, list) { 56121941Sphk if (sc->ptr == NULL) 57121941Sphk continue; 58121941Sphk if (sc->count > 0) { 59121941Sphk sc->count--; 60121941Sphk continue; 61121941Sphk } 62128678Sphk if (*sc->ptr == '.') { 63128678Sphk sc->ptr = NULL; 64247008Smav blinkers--; 65128678Sphk continue; 66140966Sphk } else if (*sc->ptr == 'U' || *sc->ptr == 'u') { 67140966Sphk if (sc->last_second == time_second) 68140966Sphk continue; 69140966Sphk sc->last_second = time_second; 70140966Sphk sc->func(sc->private, *sc->ptr == 'U'); 71128678Sphk } else if (*sc->ptr >= 'a' && *sc->ptr <= 'j') { 72123494Sphk sc->func(sc->private, 0); 73140966Sphk sc->count = (*sc->ptr & 0xf) - 1; 74128678Sphk } else if (*sc->ptr >= 'A' && *sc->ptr <= 'J') { 75123494Sphk sc->func(sc->private, 1); 76140966Sphk sc->count = (*sc->ptr & 0xf) - 1; 77128678Sphk } 78123494Sphk sc->ptr++; 79123494Sphk if (*sc->ptr == '\0') 80121941Sphk sc->ptr = sc->str; 81121941Sphk } 82247008Smav if (blinkers > 0) 83247008Smav callout_reset(&led_ch, hz / 10, led_timeout, p); 84121941Sphk} 85121941Sphk 86121941Sphkstatic int 87219951Smavled_state(struct ledsc *sc, struct sbuf **sb, int state) 88140966Sphk{ 89140966Sphk struct sbuf *sb2 = NULL; 90140966Sphk 91219951Smav sb2 = sc->spec; 92219951Smav sc->spec = *sb; 93219951Smav if (*sb != NULL) { 94219951Smav sc->str = sbuf_data(*sb); 95247008Smav if (sc->ptr == NULL) { 96247008Smav blinkers++; 97247008Smav callout_reset(&led_ch, hz / 10, led_timeout, NULL); 98247008Smav } 99219951Smav sc->ptr = sc->str; 100219951Smav } else { 101219951Smav sc->str = NULL; 102247008Smav if (sc->ptr != NULL) 103247008Smav blinkers--; 104219951Smav sc->ptr = NULL; 105219951Smav sc->func(sc->private, state); 106140966Sphk } 107219951Smav sc->count = 0; 108219951Smav *sb = sb2; 109140966Sphk return(0); 110140966Sphk} 111140966Sphk 112140966Sphkstatic int 113219951Smavled_parse(const char *s, struct sbuf **sb, int *state) 114121941Sphk{ 115219951Smav int i, error; 116121941Sphk 117121941Sphk /* 118121941Sphk * Handle "on" and "off" immediately so people can flash really 119121941Sphk * fast from userland if they want to 120121941Sphk */ 121121941Sphk if (*s == '0' || *s == '1') { 122219951Smav *state = *s & 1; 123219951Smav return (0); 124121941Sphk } 125121941Sphk 126219951Smav *state = 0; 127219951Smav *sb = sbuf_new_auto(); 128219951Smav if (*sb == NULL) 129121941Sphk return (ENOMEM); 130121941Sphk switch(s[0]) { 131121941Sphk /* 132121941Sphk * Flash, default is 100msec/100msec. 133121941Sphk * 'f2' sets 200msec/200msec etc. 134121941Sphk */ 135121941Sphk case 'f': 136121941Sphk if (s[1] >= '1' && s[1] <= '9') 137121941Sphk i = s[1] - '1'; 138121941Sphk else 139121941Sphk i = 0; 140219951Smav sbuf_printf(*sb, "%c%c", 'A' + i, 'a' + i); 141121941Sphk break; 142121941Sphk /* 143121941Sphk * Digits, flashes out numbers. 144121941Sphk * 'd12' becomes -__________-_-______________________________ 145121941Sphk */ 146121941Sphk case 'd': 147121941Sphk for(s++; *s; s++) { 148121941Sphk if (!isdigit(*s)) 149121941Sphk continue; 150121941Sphk i = *s - '0'; 151121941Sphk if (i == 0) 152121941Sphk i = 10; 153121941Sphk for (; i > 1; i--) 154219951Smav sbuf_cat(*sb, "Aa"); 155219951Smav sbuf_cat(*sb, "Aj"); 156121941Sphk } 157219951Smav sbuf_cat(*sb, "jj"); 158121941Sphk break; 159121941Sphk /* 160121941Sphk * String, roll your own. 161123014Sphk * 'a-j' gives "off" for n/10 sec. 162123014Sphk * 'A-J' gives "on" for n/10 sec. 163121941Sphk * no delay before repeat 164121941Sphk * 'sAaAbBa' becomes _-_--__- 165121941Sphk */ 166121941Sphk case 's': 167121941Sphk for(s++; *s; s++) { 168128678Sphk if ((*s >= 'a' && *s <= 'j') || 169128678Sphk (*s >= 'A' && *s <= 'J') || 170140966Sphk *s == 'U' || *s <= 'u' || 171128678Sphk *s == '.') 172219951Smav sbuf_bcat(*sb, s, 1); 173121941Sphk } 174121941Sphk break; 175121941Sphk /* 176121941Sphk * Morse. 177121941Sphk * '.' becomes _- 178121941Sphk * '-' becomes _--- 179121941Sphk * ' ' becomes __ 180121941Sphk * '\n' becomes ____ 181121941Sphk * 1sec pause between repeats 182121941Sphk * '... --- ...' -> _-_-_-___---_---_---___-_-_-__________ 183121941Sphk */ 184121941Sphk case 'm': 185121941Sphk for(s++; *s; s++) { 186121941Sphk if (*s == '.') 187219951Smav sbuf_cat(*sb, "aA"); 188121941Sphk else if (*s == '-') 189219951Smav sbuf_cat(*sb, "aC"); 190121941Sphk else if (*s == ' ') 191219951Smav sbuf_cat(*sb, "b"); 192121956Sphk else if (*s == '\n') 193219951Smav sbuf_cat(*sb, "d"); 194121941Sphk } 195219951Smav sbuf_cat(*sb, "j"); 196121941Sphk break; 197121941Sphk default: 198219951Smav sbuf_delete(*sb); 199123494Sphk return (EINVAL); 200121941Sphk } 201219951Smav error = sbuf_finish(*sb); 202219951Smav if (error != 0 || sbuf_len(*sb) == 0) { 203219951Smav *sb = NULL; 204212371Smdf return (error); 205121941Sphk } 206219951Smav return (0); 207219951Smav} 208121941Sphk 209219951Smavstatic int 210219951Smavled_write(struct cdev *dev, struct uio *uio, int ioflag) 211219951Smav{ 212219951Smav struct ledsc *sc; 213219951Smav char *s; 214219951Smav struct sbuf *sb = NULL; 215219951Smav int error, state = 0; 216219951Smav 217219951Smav if (uio->uio_resid > 512) 218219951Smav return (EINVAL); 219219951Smav s = malloc(uio->uio_resid + 1, M_DEVBUF, M_WAITOK); 220219951Smav s[uio->uio_resid] = '\0'; 221219951Smav error = uiomove(s, uio->uio_resid, uio); 222219951Smav if (error) { 223219951Smav free(s, M_DEVBUF); 224219951Smav return (error); 225219951Smav } 226219951Smav error = led_parse(s, &sb, &state); 227219951Smav free(s, M_DEVBUF); 228219951Smav if (error) 229219951Smav return (error); 230219951Smav mtx_lock(&led_mtx); 231219951Smav sc = dev->si_drv1; 232219951Smav if (sc != NULL) 233219951Smav error = led_state(sc, &sb, state); 234219951Smav mtx_unlock(&led_mtx); 235219951Smav if (sb != NULL) 236219951Smav sbuf_delete(sb); 237219951Smav return (error); 238121941Sphk} 239121941Sphk 240219951Smavint 241219951Smavled_set(char const *name, char const *cmd) 242219951Smav{ 243219951Smav struct ledsc *sc; 244219951Smav struct sbuf *sb = NULL; 245219951Smav int error, state = 0; 246219951Smav 247219951Smav error = led_parse(cmd, &sb, &state); 248219951Smav if (error) 249219951Smav return (error); 250219951Smav mtx_lock(&led_mtx); 251219951Smav LIST_FOREACH(sc, &led_list, list) { 252219951Smav if (strcmp(sc->name, name) == 0) 253219951Smav break; 254219951Smav } 255219951Smav if (sc != NULL) 256219951Smav error = led_state(sc, &sb, state); 257219951Smav else 258219951Smav error = ENOENT; 259219951Smav mtx_unlock(&led_mtx); 260219951Smav if (sb != NULL) 261219951Smav sbuf_delete(sb); 262219951Smav return (0); 263219951Smav} 264219951Smav 265121941Sphkstatic struct cdevsw led_cdevsw = { 266126080Sphk .d_version = D_VERSION, 267125810Sphk .d_write = led_write, 268125810Sphk .d_name = "LED", 269121941Sphk}; 270121941Sphk 271130585Sphkstruct cdev * 272121941Sphkled_create(led_t *func, void *priv, char const *name) 273121941Sphk{ 274168974Sphk 275168974Sphk return (led_create_state(func, priv, name, 0)); 276168974Sphk} 277168974Sphkstruct cdev * 278168974Sphkled_create_state(led_t *func, void *priv, char const *name, int state) 279168974Sphk{ 280121941Sphk struct ledsc *sc; 281121941Sphk 282140966Sphk sc = malloc(sizeof *sc, M_LED, M_WAITOK | M_ZERO); 283121941Sphk 284140966Sphk sx_xlock(&led_sx); 285219951Smav sc->name = strdup(name, M_LED); 286140966Sphk sc->unit = alloc_unr(led_unit); 287121941Sphk sc->private = priv; 288121941Sphk sc->func = func; 289183381Sed sc->dev = make_dev(&led_cdevsw, sc->unit, 290140966Sphk UID_ROOT, GID_WHEEL, 0600, "led/%s", name); 291140966Sphk sx_xunlock(&led_sx); 292140966Sphk 293140966Sphk mtx_lock(&led_mtx); 294121941Sphk sc->dev->si_drv1 = sc; 295121941Sphk LIST_INSERT_HEAD(&led_list, sc, list); 296168974Sphk sc->func(sc->private, state != 0); 297121941Sphk mtx_unlock(&led_mtx); 298140966Sphk 299121941Sphk return (sc->dev); 300121941Sphk} 301121941Sphk 302121941Sphkvoid 303130585Sphkled_destroy(struct cdev *dev) 304121941Sphk{ 305121941Sphk struct ledsc *sc; 306121941Sphk 307140966Sphk mtx_lock(&led_mtx); 308121941Sphk sc = dev->si_drv1; 309140966Sphk dev->si_drv1 = NULL; 310247008Smav if (sc->ptr != NULL) 311247008Smav blinkers--; 312121941Sphk LIST_REMOVE(sc, list); 313140966Sphk if (LIST_EMPTY(&led_list)) 314140966Sphk callout_stop(&led_ch); 315121941Sphk mtx_unlock(&led_mtx); 316140966Sphk 317140966Sphk sx_xlock(&led_sx); 318140966Sphk free_unr(led_unit, sc->unit); 319140966Sphk destroy_dev(dev); 320122963Sphk if (sc->spec != NULL) 321122963Sphk sbuf_delete(sc->spec); 322219951Smav free(sc->name, M_LED); 323121941Sphk free(sc, M_LED); 324140966Sphk sx_xunlock(&led_sx); 325121941Sphk} 326140966Sphk 327140966Sphkstatic void 328140966Sphkled_drvinit(void *unused) 329140966Sphk{ 330140966Sphk 331179413Sed led_unit = new_unrhdr(0, INT_MAX, NULL); 332140966Sphk mtx_init(&led_mtx, "LED mtx", NULL, MTX_DEF); 333140966Sphk sx_init(&led_sx, "LED sx"); 334247008Smav callout_init_mtx(&led_ch, &led_mtx, 0); 335140966Sphk} 336140966Sphk 337140966SphkSYSINIT(leddev, SI_SUB_DRIVERS, SI_ORDER_MIDDLE, led_drvinit, NULL); 338