Merge branch 'v2021.07-rc1' of https://github.com/lftan/u-boot
[platform/kernel/u-boot.git] / cmd / legacy_led.c
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
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  */
84 void __weak __led_blink(led_id_t mask, int freq)
85 {
86 }
87
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 );