diff --git a/commands/timeout.c b/commands/timeout.c index c8e930c..2b99d4f 100644 --- a/commands/timeout.c +++ b/commands/timeout.c @@ -16,40 +16,35 @@ * GNU General Public License for more details. * */ -#include + #include -#include #include #include -#include #include +#include -#define TIMEOUT_RETURN (1 << 0) -#define TIMEOUT_CTRLC (1 << 1) -#define TIMEOUT_ANYKEY (1 << 2) -#define TIMEOUT_SILENT (1 << 3) +#include static int do_timeout(int argc, char *argv[]) { - int timeout = 3, ret = 1; - int flags = 0, opt, countdown; - int key = 0; - uint64_t start, second; + int timeout, ret, opt; + unsigned flags = 0; + char str[2] = { }; const char *varname = NULL; while((opt = getopt(argc, argv, "crsav:")) > 0) { switch(opt) { case 'r': - flags |= TIMEOUT_RETURN; + flags |= CONSOLE_COUNTDOWN_RETURN; break; case 'c': - flags |= TIMEOUT_CTRLC; + flags |= CONSOLE_COUNTDOWN_CTRLC; break; case 'a': - flags |= TIMEOUT_ANYKEY; + flags |= CONSOLE_COUNTDOWN_ANYKEY; break; case 's': - flags |= TIMEOUT_SILENT; + flags |= CONSOLE_COUNTDOWN_SILENT; break; case 'v': varname = optarg; @@ -63,43 +58,12 @@ return COMMAND_ERROR_USAGE; timeout = simple_strtoul(argv[optind], NULL, 0); + ret = console_countdown(timeout, flags, str); - start = get_time_ns(); - second = start; - - countdown = timeout; - - if (!(flags & TIMEOUT_SILENT)) - printf("%2d", countdown--); - - do { - if (tstc()) { - key = getc(); - if (flags & TIMEOUT_CTRLC && key == 3) - goto out; - if (flags & TIMEOUT_ANYKEY) - goto out; - if (flags & TIMEOUT_RETURN && key == '\n') - goto out; - key = 0; - } - if (!(flags & TIMEOUT_SILENT) && is_timeout(second, SECOND)) { - printf("\b\b%2d", countdown--); - second += SECOND; - } - } while (!is_timeout(start, timeout * SECOND)); - - ret = 0; -out: - if (varname && key) { - char str[2] = { }; - str[0] = key; + if (varname && str[0]) setenv(varname, str); - } - if (!(flags & TIMEOUT_SILENT)) - printf("\n"); - return ret; + return ret ? 1 : 0; } BAREBOX_CMD_HELP_START(timeout) diff --git a/common/Makefile b/common/Makefile index eca1e35..2738238 100644 --- a/common/Makefile +++ b/common/Makefile @@ -20,6 +20,7 @@ obj-$(CONFIG_COMMAND_SUPPORT) += command.o obj-$(CONFIG_CONSOLE_FULL) += console.o obj-$(CONFIG_CONSOLE_SIMPLE) += console_simple.o +obj-y += console_countdown.o obj-$(CONFIG_DDR_SPD) += ddr_spd.o obj-$(CONFIG_ENV_HANDLING) += environment.o obj-$(CONFIG_ENVIRONMENT_VARIABLES) += env.o diff --git a/common/console_countdown.c b/common/console_countdown.c new file mode 100644 index 0000000..ffbdb4f --- /dev/null +++ b/common/console_countdown.c @@ -0,0 +1,67 @@ +/* + * console_countdown - contdown on the console - interruptible by a keypress + * + * Copyright (c) 2007 Sascha Hauer , Pengutronix + * + * See file CREDITS for list of people who contributed to this + * project. + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License version 2 + * as published by the Free Software Foundation. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + */ + +#include +#include +#include +#include +#include + +int console_countdown(int timeout_s, unsigned flags, char *out_key) +{ + uint64_t start, second; + int countdown, ret = -EINTR; + int key = 0; + + start = get_time_ns(); + second = start; + + countdown = timeout_s; + + if (!(flags & CONSOLE_COUNTDOWN_SILENT)) + printf("%2d", countdown--); + + do { + if (tstc()) { + key = getc(); + if (flags & CONSOLE_COUNTDOWN_ANYKEY) + goto out; + if (flags & CONSOLE_COUNTDOWN_RETURN && key == '\n') + goto out; + if (flags & CONSOLE_COUNTDOWN_CTRLC && key == 3) + goto out; + key = 0; + } + if (!(flags & CONSOLE_COUNTDOWN_SILENT) && + is_timeout(second, SECOND)) { + printf("\b\b%2d", countdown--); + second += SECOND; + } + } while (!is_timeout(start, timeout_s * SECOND)); + + ret = 0; + + out: + if (!(flags & CONSOLE_COUNTDOWN_SILENT)) + printf("\n"); + if (key && out_key) + *out_key = key; + + return ret; +} diff --git a/include/console_countdown.h b/include/console_countdown.h new file mode 100644 index 0000000..cb46964 --- /dev/null +++ b/include/console_countdown.h @@ -0,0 +1,11 @@ +#ifndef __CONSOLE_COUNTDOWN_H +#define __CONSOLE_COUNTDOWN_H + +#define CONSOLE_COUNTDOWN_SILENT (1 << 0) +#define CONSOLE_COUNTDOWN_ANYKEY (1 << 1) +#define CONSOLE_COUNTDOWN_RETURN (1 << 3) +#define CONSOLE_COUNTDOWN_CTRLC (1 << 4) + +int console_countdown(int timeout_s, unsigned flags, char *out_key); + +#endif /* __CONSOLE_COUNTDOWN_H */