[u-boot.git] / cmd / led.c

/*
 * Copyright (c) 2017 Google, Inc
 * Written by Simon Glass <[email protected]>
 *
 * SPDX-License-Identifier:     GPL-2.0+
 */

#include <common.h>
#include <command.h>
#include <dm.h>
#include <led.h>
#include <dm/uclass-internal.h>

#define LED_TOGGLE LEDST_COUNT

static const char *const state_label[] = {
	[LEDST_OFF]	= "off",
	[LEDST_ON]	= "on",
	[LEDST_TOGGLE]	= "toggle",
#ifdef CONFIG_LED_BLINK
	[LEDST_BLINK]	= "blink",
#endif
};

enum led_state_t get_led_cmd(char *var)
{
	int i;

	for (i = 0; i < LEDST_COUNT; i++) {
		if (!strncmp(var, state_label[i], strlen(var)))
			return i;
	}

	return -1;
}

static int show_led_state(struct udevice *dev)
{
	int ret;

	ret = led_get_state(dev);
	if (ret >= LEDST_COUNT)
		ret = -EINVAL;
	if (ret >= 0)
		printf("%s\n", state_label[ret]);

	return ret;
}

static int list_leds(void)
{
	struct udevice *dev;
	int ret;

	for (uclass_find_first_device(UCLASS_LED, &dev);
	     dev;
	     uclass_find_next_device(&dev)) {
		struct led_uc_plat *plat = dev_get_uclass_platdata(dev);

		if (!plat->label)
			continue;
		printf("%-15s ", plat->label);
		if (device_active(dev)) {
			ret = show_led_state(dev);
			if (ret < 0)
				printf("Error %d\n", ret);
		} else {
			printf("<inactive>\n");
		}
	}

	return 0;
}

int do_led(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	enum led_state_t cmd;
	const char *led_label;
	struct udevice *dev;
#ifdef CONFIG_LED_BLINK
	int freq_ms = 0;
#endif
	int ret;

	/* Validate arguments */
	if (argc < 2)
		return CMD_RET_USAGE;
	led_label = argv[1];
	if (*led_label == 'l')
		return list_leds();

	cmd = argc > 2 ? get_led_cmd(argv[2]) : LEDST_COUNT;
#ifdef CONFIG_LED_BLINK
	if (cmd == LEDST_BLINK) {
		if (argc < 4)
			return CMD_RET_USAGE;
		freq_ms = simple_strtoul(argv[3], NULL, 10);
	}
#endif
	ret = led_get_by_label(led_label, &dev);
	if (ret) {
		printf("LED '%s' not found (err=%d)\n", led_label, ret);
		return CMD_RET_FAILURE;
	}
	switch (cmd) {
	case LEDST_OFF:
	case LEDST_ON:
	case LEDST_TOGGLE:
		ret = led_set_state(dev, cmd);
		break;
#ifdef CONFIG_LED_BLINK
	case LEDST_BLINK:
		ret = led_set_period(dev, freq_ms);
		if (!ret)
			ret = led_set_state(dev, LEDST_BLINK);
		break;
#endif
	case LEDST_COUNT:
		printf("LED '%s': ", led_label);
		ret = show_led_state(dev);
		break;
	}
	if (ret < 0) {
		printf("LED '%s' operation failed (err=%d)\n", led_label, ret);
		return CMD_RET_FAILURE;
	}

	return 0;
}

#ifdef CONFIG_LED_BLINK
#define BLINK "|blink [blink-freq in ms]"
#else
#define BLINK ""
#endif

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