1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * (C) Copyright 2010
4 * Jason Kridner <jkridner@beagleboard.org>
5 *
6 * Based on cmd_led.c patch from:
7 * http://www.mail-archive.com/u-boot@lists.denx.de/msg06873.html
8 * (C) Copyright 2008
9 * Ulf Samuelsson <ulf.samuelsson@atmel.com>
10 */
11
12 #include <common.h>
13 #include <config.h>
14 #include <command.h>
15 #include <status_led.h>
16
17 struct led_tbl_s {
18 char *string; /* String for use in the command */
19 led_id_t mask; /* Mask used for calling __led_set() */
20 void (*off)(void); /* Optional function for turning LED off */
21 void (*on)(void); /* Optional function for turning LED on */
22 void (*toggle)(void);/* Optional function for toggling LED */
23 };
24
25 typedef struct led_tbl_s led_tbl_t;
26
27 static const led_tbl_t led_commands[] = {
28 #ifdef CONFIG_LED_STATUS_BOARD_SPECIFIC
29 #ifdef CONFIG_LED_STATUS0
30 { "0", CONFIG_LED_STATUS_BIT, NULL, NULL, NULL },
31 #endif
32 #ifdef CONFIG_LED_STATUS1
33 { "1", CONFIG_LED_STATUS_BIT1, NULL, NULL, NULL },
34 #endif
35 #ifdef CONFIG_LED_STATUS2
36 { "2", CONFIG_LED_STATUS_BIT2, NULL, NULL, NULL },
37 #endif
38 #ifdef CONFIG_LED_STATUS3
39 { "3", CONFIG_LED_STATUS_BIT3, NULL, NULL, NULL },
40 #endif
41 #ifdef CONFIG_LED_STATUS4
42 { "4", CONFIG_LED_STATUS_BIT4, NULL, NULL, NULL },
43 #endif
44 #ifdef CONFIG_LED_STATUS5
45 { "5", CONFIG_LED_STATUS_BIT5, NULL, NULL, NULL },
46 #endif
47 #endif
48 #ifdef CONFIG_LED_STATUS_GREEN
49 { "green", CONFIG_LED_STATUS_GREEN, green_led_off, green_led_on, NULL },
50 #endif
51 #ifdef CONFIG_LED_STATUS_YELLOW
52 { "yellow", CONFIG_LED_STATUS_YELLOW, yellow_led_off, yellow_led_on,
53 NULL },
54 #endif
55 #ifdef CONFIG_LED_STATUS_RED
56 { "red", CONFIG_LED_STATUS_RED, red_led_off, red_led_on, NULL },
57 #endif
58 #ifdef CONFIG_LED_STATUS_BLUE
59 { "blue", CONFIG_LED_STATUS_BLUE, blue_led_off, blue_led_on, NULL },
60 #endif
61 { NULL, 0, NULL, NULL, NULL }
62 };
63
64 enum led_cmd { LED_ON, LED_OFF, LED_TOGGLE, LED_BLINK };
65
get_led_cmd(char * var)66 enum led_cmd get_led_cmd(char *var)
67 {
68 if (strcmp(var, "off") == 0)
69 return LED_OFF;
70 if (strcmp(var, "on") == 0)
71 return LED_ON;
72 if (strcmp(var, "toggle") == 0)
73 return LED_TOGGLE;
74 if (strcmp(var, "blink") == 0)
75 return LED_BLINK;
76
77 return -1;
78 }
79
80 /*
81 * LED drivers providing a blinking LED functionality, like the
82 * PCA9551, can override this empty weak function
83 */
__led_blink(led_id_t mask,int freq)84 void __weak __led_blink(led_id_t mask, int freq)
85 {
86 }
87
do_legacy_led(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])88 int do_legacy_led(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
89 {
90 int i, match = 0;
91 enum led_cmd cmd;
92 int freq;
93
94 /* Validate arguments */
95 if ((argc < 3) || (argc > 4))
96 return CMD_RET_USAGE;
97
98 cmd = get_led_cmd(argv[2]);
99 if (cmd < 0) {
100 return CMD_RET_USAGE;
101 }
102
103 for (i = 0; led_commands[i].string; i++) {
104 if ((strcmp("all", argv[1]) == 0) ||
105 (strcmp(led_commands[i].string, argv[1]) == 0)) {
106 match = 1;
107 switch (cmd) {
108 case LED_ON:
109 if (led_commands[i].on)
110 led_commands[i].on();
111 else
112 __led_set(led_commands[i].mask,
113 CONFIG_LED_STATUS_ON);
114 break;
115 case LED_OFF:
116 if (led_commands[i].off)
117 led_commands[i].off();
118 else
119 __led_set(led_commands[i].mask,
120 CONFIG_LED_STATUS_OFF);
121 break;
122 case LED_TOGGLE:
123 if (led_commands[i].toggle)
124 led_commands[i].toggle();
125 else
126 __led_toggle(led_commands[i].mask);
127 break;
128 case LED_BLINK:
129 if (argc != 4)
130 return CMD_RET_USAGE;
131
132 freq = simple_strtoul(argv[3], NULL, 10);
133 __led_blink(led_commands[i].mask, freq);
134 }
135 /* Need to set only 1 led if led_name wasn't 'all' */
136 if (strcmp("all", argv[1]) != 0)
137 break;
138 }
139 }
140
141 /* If we ran out of matches, print Usage */
142 if (!match) {
143 return CMD_RET_USAGE;
144 }
145
146 return 0;
147 }
148
149 U_BOOT_CMD(
150 led, 4, 1, do_legacy_led,
151 "["
152 #ifdef CONFIG_LED_STATUS_BOARD_SPECIFIC
153 #ifdef CONFIG_LED_STATUS0
154 "0|"
155 #endif
156 #ifdef CONFIG_LED_STATUS1
157 "1|"
158 #endif
159 #ifdef CONFIG_LED_STATUS2
160 "2|"
161 #endif
162 #ifdef CONFIG_LED_STATUS3
163 "3|"
164 #endif
165 #ifdef CONFIG_LED_STATUS4
166 "4|"
167 #endif
168 #ifdef CONFIG_LED_STATUS5
169 "5|"
170 #endif
171 #endif
172 #ifdef CONFIG_LED_STATUS_GREEN
173 "green|"
174 #endif
175 #ifdef CONFIG_LED_STATUS_YELLOW
176 "yellow|"
177 #endif
178 #ifdef CONFIG_LED_STATUS_RED
179 "red|"
180 #endif
181 #ifdef CONFIG_LED_STATUS_BLUE
182 "blue|"
183 #endif
184 "all] [on|off|toggle|blink] [blink-freq in ms]",
185 "[led_name] [on|off|toggle|blink] sets or clears led(s)"
186 );
187