1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun * Copyright (c) 2017 Google, Inc
3*4882a593Smuzhiyun * Written by Simon Glass <sjg@chromium.org>
4*4882a593Smuzhiyun *
5*4882a593Smuzhiyun * SPDX-License-Identifier: GPL-2.0+
6*4882a593Smuzhiyun */
7*4882a593Smuzhiyun
8*4882a593Smuzhiyun #include <common.h>
9*4882a593Smuzhiyun #include <command.h>
10*4882a593Smuzhiyun #include <dm.h>
11*4882a593Smuzhiyun #include <led.h>
12*4882a593Smuzhiyun #include <dm/uclass-internal.h>
13*4882a593Smuzhiyun
14*4882a593Smuzhiyun #define LED_TOGGLE LEDST_COUNT
15*4882a593Smuzhiyun
16*4882a593Smuzhiyun static const char *const state_label[] = {
17*4882a593Smuzhiyun [LEDST_OFF] = "off",
18*4882a593Smuzhiyun [LEDST_ON] = "on",
19*4882a593Smuzhiyun [LEDST_TOGGLE] = "toggle",
20*4882a593Smuzhiyun #ifdef CONFIG_LED_BLINK
21*4882a593Smuzhiyun [LEDST_BLINK] = "blink",
22*4882a593Smuzhiyun #endif
23*4882a593Smuzhiyun };
24*4882a593Smuzhiyun
get_led_cmd(char * var)25*4882a593Smuzhiyun enum led_state_t get_led_cmd(char *var)
26*4882a593Smuzhiyun {
27*4882a593Smuzhiyun int i;
28*4882a593Smuzhiyun
29*4882a593Smuzhiyun for (i = 0; i < LEDST_COUNT; i++) {
30*4882a593Smuzhiyun if (!strncmp(var, state_label[i], strlen(var)))
31*4882a593Smuzhiyun return i;
32*4882a593Smuzhiyun }
33*4882a593Smuzhiyun
34*4882a593Smuzhiyun return -1;
35*4882a593Smuzhiyun }
36*4882a593Smuzhiyun
show_led_state(struct udevice * dev)37*4882a593Smuzhiyun static int show_led_state(struct udevice *dev)
38*4882a593Smuzhiyun {
39*4882a593Smuzhiyun int ret;
40*4882a593Smuzhiyun
41*4882a593Smuzhiyun ret = led_get_state(dev);
42*4882a593Smuzhiyun if (ret >= LEDST_COUNT)
43*4882a593Smuzhiyun ret = -EINVAL;
44*4882a593Smuzhiyun if (ret >= 0)
45*4882a593Smuzhiyun printf("%s\n", state_label[ret]);
46*4882a593Smuzhiyun
47*4882a593Smuzhiyun return ret;
48*4882a593Smuzhiyun }
49*4882a593Smuzhiyun
list_leds(void)50*4882a593Smuzhiyun static int list_leds(void)
51*4882a593Smuzhiyun {
52*4882a593Smuzhiyun struct udevice *dev;
53*4882a593Smuzhiyun int ret;
54*4882a593Smuzhiyun
55*4882a593Smuzhiyun for (uclass_find_first_device(UCLASS_LED, &dev);
56*4882a593Smuzhiyun dev;
57*4882a593Smuzhiyun uclass_find_next_device(&dev)) {
58*4882a593Smuzhiyun struct led_uc_plat *plat = dev_get_uclass_platdata(dev);
59*4882a593Smuzhiyun
60*4882a593Smuzhiyun if (!plat->label)
61*4882a593Smuzhiyun continue;
62*4882a593Smuzhiyun printf("%-15s ", plat->label);
63*4882a593Smuzhiyun if (device_active(dev)) {
64*4882a593Smuzhiyun ret = show_led_state(dev);
65*4882a593Smuzhiyun if (ret < 0)
66*4882a593Smuzhiyun printf("Error %d\n", ret);
67*4882a593Smuzhiyun } else {
68*4882a593Smuzhiyun printf("<inactive>\n");
69*4882a593Smuzhiyun }
70*4882a593Smuzhiyun }
71*4882a593Smuzhiyun
72*4882a593Smuzhiyun return 0;
73*4882a593Smuzhiyun }
74*4882a593Smuzhiyun
do_led(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])75*4882a593Smuzhiyun int do_led(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
76*4882a593Smuzhiyun {
77*4882a593Smuzhiyun enum led_state_t cmd;
78*4882a593Smuzhiyun const char *led_label;
79*4882a593Smuzhiyun struct udevice *dev;
80*4882a593Smuzhiyun #ifdef CONFIG_LED_BLINK
81*4882a593Smuzhiyun int freq_ms = 0;
82*4882a593Smuzhiyun #endif
83*4882a593Smuzhiyun int ret;
84*4882a593Smuzhiyun
85*4882a593Smuzhiyun /* Validate arguments */
86*4882a593Smuzhiyun if (argc < 2)
87*4882a593Smuzhiyun return CMD_RET_USAGE;
88*4882a593Smuzhiyun led_label = argv[1];
89*4882a593Smuzhiyun if (*led_label == 'l')
90*4882a593Smuzhiyun return list_leds();
91*4882a593Smuzhiyun
92*4882a593Smuzhiyun cmd = argc > 2 ? get_led_cmd(argv[2]) : LEDST_COUNT;
93*4882a593Smuzhiyun #ifdef CONFIG_LED_BLINK
94*4882a593Smuzhiyun if (cmd == LEDST_BLINK) {
95*4882a593Smuzhiyun if (argc < 4)
96*4882a593Smuzhiyun return CMD_RET_USAGE;
97*4882a593Smuzhiyun freq_ms = simple_strtoul(argv[3], NULL, 10);
98*4882a593Smuzhiyun }
99*4882a593Smuzhiyun #endif
100*4882a593Smuzhiyun ret = led_get_by_label(led_label, &dev);
101*4882a593Smuzhiyun if (ret) {
102*4882a593Smuzhiyun printf("LED '%s' not found (err=%d)\n", led_label, ret);
103*4882a593Smuzhiyun return CMD_RET_FAILURE;
104*4882a593Smuzhiyun }
105*4882a593Smuzhiyun switch (cmd) {
106*4882a593Smuzhiyun case LEDST_OFF:
107*4882a593Smuzhiyun case LEDST_ON:
108*4882a593Smuzhiyun case LEDST_TOGGLE:
109*4882a593Smuzhiyun ret = led_set_state(dev, cmd);
110*4882a593Smuzhiyun break;
111*4882a593Smuzhiyun #ifdef CONFIG_LED_BLINK
112*4882a593Smuzhiyun case LEDST_BLINK:
113*4882a593Smuzhiyun ret = led_set_period(dev, freq_ms);
114*4882a593Smuzhiyun if (!ret)
115*4882a593Smuzhiyun ret = led_set_state(dev, LEDST_BLINK);
116*4882a593Smuzhiyun break;
117*4882a593Smuzhiyun #endif
118*4882a593Smuzhiyun case LEDST_COUNT:
119*4882a593Smuzhiyun printf("LED '%s': ", led_label);
120*4882a593Smuzhiyun ret = show_led_state(dev);
121*4882a593Smuzhiyun break;
122*4882a593Smuzhiyun }
123*4882a593Smuzhiyun if (ret < 0) {
124*4882a593Smuzhiyun printf("LED '%s' operation failed (err=%d)\n", led_label, ret);
125*4882a593Smuzhiyun return CMD_RET_FAILURE;
126*4882a593Smuzhiyun }
127*4882a593Smuzhiyun
128*4882a593Smuzhiyun return 0;
129*4882a593Smuzhiyun }
130*4882a593Smuzhiyun
131*4882a593Smuzhiyun #ifdef CONFIG_LED_BLINK
132*4882a593Smuzhiyun #define BLINK "|blink [blink-freq in ms]"
133*4882a593Smuzhiyun #else
134*4882a593Smuzhiyun #define BLINK ""
135*4882a593Smuzhiyun #endif
136*4882a593Smuzhiyun
137*4882a593Smuzhiyun U_BOOT_CMD(
138*4882a593Smuzhiyun led, 4, 1, do_led,
139*4882a593Smuzhiyun "manage LEDs",
140*4882a593Smuzhiyun "<led_label> on|off|toggle" BLINK "\tChange LED state\n"
141*4882a593Smuzhiyun "led [<led_label>\tGet LED state\n"
142*4882a593Smuzhiyun "led list\t\tshow a list of LEDs"
143*4882a593Smuzhiyun );
144