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

// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright (c) 2017 Google, Inc
 * Written by Simon Glass <sjg@chromium.org>
 */

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