common/cmd_led.c

   1 /*
   2  * (C) Copyright 2010
   3  * Jason Kridner <jkridner@beagleboard.org>
   4  *
   5  * Based on cmd_led.c patch from:
   6  * http://www.mail-archive.com/u-boot@lists.denx.de/msg06873.html
   7  * (C) Copyright 2008
   8  * Ulf Samuelsson <ulf.samuelsson@atmel.com>
   9  *
  10  * See file CREDITS for list of people who contributed to this
  11  * project.
  12  *
  13  * This program is free software; you can redistribute it and/or
  14  * modify it under the terms of the GNU General Public License as
  15  * published by the Free Software Foundation; either version 2 of
  16  * the License, or (at your option) any later version.
  17  *
  18  * This program is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU General Public License for more details.
  22  *
  23  * You should have received a copy of the GNU General Public License
  24  * along with this program; if not, write to the Free Software
  25  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
  26  * MA 02111-1307 USA
  27  */
  28
  29 #include <common.h>
  30 #include <config.h>
  31 #include <command.h>
  32 #include <status_led.h>
  33
  34 struct led_tbl_s {
  35         char            *string;        /* String for use in the command */
  36         led_id_t        mask;           /* Mask used for calling __led_set() */
  37         void            (*off)(void);   /* Optional function for turning LED off */
  38         void            (*on)(void);    /* Optional function for turning LED on */
  39         void            (*toggle)(void);/* Optional function for toggling LED */
  40 };
  41
  42 typedef struct led_tbl_s led_tbl_t;
  43
  44 static const led_tbl_t led_commands[] = {
  45 #ifdef CONFIG_BOARD_SPECIFIC_LED
  46 #ifdef STATUS_LED_BIT
  47         { "0", STATUS_LED_BIT, NULL, NULL, NULL },
  48 #endif
  49 #ifdef STATUS_LED_BIT1
  50         { "1", STATUS_LED_BIT1, NULL, NULL, NULL },
  51 #endif
  52 #ifdef STATUS_LED_BIT2
  53         { "2", STATUS_LED_BIT2, NULL, NULL, NULL },
  54 #endif
  55 #ifdef STATUS_LED_BIT3
  56         { "3", STATUS_LED_BIT3, NULL, NULL, NULL },
  57 #endif
  58 #endif
  59 #ifdef STATUS_LED_GREEN
  60         { "green", STATUS_LED_GREEN, green_led_off, green_led_on, NULL },
  61 #endif
  62 #ifdef STATUS_LED_YELLOW
  63         { "yellow", STATUS_LED_YELLOW, yellow_led_off, yellow_led_on, NULL },
  64 #endif
  65 #ifdef STATUS_LED_RED
  66         { "red", STATUS_LED_RED, red_led_off, red_led_on, NULL },
  67 #endif
  68 #ifdef STATUS_LED_BLUE
  69         { "blue", STATUS_LED_BLUE, blue_led_off, blue_led_on, NULL },
  70 #endif
  71         { NULL, 0, NULL, NULL, NULL }
  72 };
  73
  74 enum led_cmd { LED_ON, LED_OFF, LED_TOGGLE };
  75
  76 enum led_cmd get_led_cmd(char *var)
  77 {
  78         if (strcmp(var, "off") == 0) {
  79                 return LED_OFF;
  80         }
  81         if (strcmp(var, "on") == 0) {
  82                 return LED_ON;
  83         }
  84         if (strcmp(var, "toggle") == 0)
  85                 return LED_TOGGLE;
  86         return -1;
  87 }
  88
  89 int do_led (cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  90 {
  91         int i, match = 0;
  92         enum led_cmd cmd;
  93
  94         /* Validate arguments */
  95         if ((argc != 3)) {
  96                 return CMD_RET_USAGE;
  97         }
  98
  99         cmd = get_led_cmd(argv[2]);
 100         if (cmd < 0) {
 101                 return CMD_RET_USAGE;
 102         }
 103
 104         for (i = 0; led_commands[i].string; i++) {
 105                 if ((strcmp("all", argv[1]) == 0) ||
 106                     (strcmp(led_commands[i].string, argv[1]) == 0)) {
 107                         match = 1;
 108                         switch (cmd) {
 109                         case LED_ON:
 110                                 if (led_commands[i].on)
 111                                         led_commands[i].on();
 112                                 else
 113                                         __led_set(led_commands[i].mask, 1);
 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, 0);
 120                                 break;
 121                         case LED_TOGGLE:
 122                                 if (led_commands[i].toggle)
 123                                         led_commands[i].toggle();
 124                                 else
 125                                         __led_toggle(led_commands[i].mask);
 126                         }
 127                         /* Need to set only 1 led if led_name wasn't 'all' */
 128                         if (strcmp("all", argv[1]) != 0)
 129                                 break;
 130                 }
 131         }
 132
 133         /* If we ran out of matches, print Usage */
 134         if (!match) {
 135                 return CMD_RET_USAGE;
 136         }
 137
 138         return 0;
 139 }
 140
 141 U_BOOT_CMD(
 142         led, 3, 1, do_led,
 143         "["
 144 #ifdef CONFIG_BOARD_SPECIFIC_LED
 145 #ifdef STATUS_LED_BIT
 146         "0|"
 147 #endif
 148 #ifdef STATUS_LED_BIT1
 149         "1|"
 150 #endif
 151 #ifdef STATUS_LED_BIT2
 152         "2|"
 153 #endif
 154 #ifdef STATUS_LED_BIT3
 155         "3|"
 156 #endif
 157 #endif
 158 #ifdef STATUS_LED_GREEN
 159         "green|"
 160 #endif
 161 #ifdef STATUS_LED_YELLOW
 162         "yellow|"
 163 #endif
 164 #ifdef STATUS_LED_RED
 165         "red|"
 166 #endif
 167 #ifdef STATUS_LED_BLUE
 168         "blue|"
 169 #endif
 170         "all] [on|off|toggle]",
 171         "[led_name] [on|off|toggle] sets or clears led(s)"
 172 );