diff mbox series

[U-Boot] bootcount: Add bootcount command

Message ID 1526104187-19896-1-git-send-email-alex.kiernan@gmail.com
State Accepted
Commit b11ed7d602a48021d8ab96518ad80996d2b86077
Delegated to: Tom Rini
Headers show
Series [U-Boot] bootcount: Add bootcount command | expand

Commit Message

Alex Kiernan May 12, 2018, 5:49 a.m. UTC
Add a command to manipulate the bootcounter. This is useful if you can
run device recovery from inside U-Boot and need to reset the bootcounter
after executing that process as part of altbootcmd.

Signed-off-by: Alex Kiernan <alex.kiernan@gmail.com>
---

 cmd/Kconfig     |  7 +++++++
 cmd/Makefile    |  1 +
 cmd/bootcount.c | 61 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 69 insertions(+)
 create mode 100644 cmd/bootcount.c

Comments

Lukasz Majewski May 13, 2018, 2:56 p.m. UTC | #1
On Sat, 12 May 2018 05:49:47 +0000
Alex Kiernan <alex.kiernan@gmail.com> wrote:

> Add a command to manipulate the bootcounter. This is useful if you can
> run device recovery from inside U-Boot and need to reset the
> bootcounter after executing that process as part of altbootcmd.
> 
> Signed-off-by: Alex Kiernan <alex.kiernan@gmail.com>
> ---
> 
>  cmd/Kconfig     |  7 +++++++
>  cmd/Makefile    |  1 +
>  cmd/bootcount.c | 61
> +++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 3 files
> changed, 69 insertions(+) create mode 100644 cmd/bootcount.c
> 
> diff --git a/cmd/Kconfig b/cmd/Kconfig
> index eb14e08..825d4ad 100644
> --- a/cmd/Kconfig
> +++ b/cmd/Kconfig
> @@ -1218,6 +1218,13 @@ config CMD_BMP
>  	  the image into RAM, then using this command to look at it
> or display it.
>  
> +config CMD_BOOTCOUNT
> +	bool "bootcount"
> +	depends on BOOTCOUNT_LIMIT
> +	help
> +	  Enable the bootcount command, which allows interrogation
> and
> +	  reset of the bootcounter.
> +
>  config CMD_BSP
>  	bool "Enable board-specific commands"
>  	help
> diff --git a/cmd/Makefile b/cmd/Makefile
> index 0d7322e..c05dc2b 100644
> --- a/cmd/Makefile
> +++ b/cmd/Makefile
> @@ -22,6 +22,7 @@ obj-$(CONFIG_CMD_BEDBUG) += bedbug.o
>  obj-$(CONFIG_CMD_BINOP) += binop.o
>  obj-$(CONFIG_CMD_BLOCK_CACHE) += blkcache.o
>  obj-$(CONFIG_CMD_BMP) += bmp.o
> +obj-$(CONFIG_CMD_BOOTCOUNT) += bootcount.o
>  obj-$(CONFIG_CMD_BOOTEFI) += bootefi.o
>  obj-$(CONFIG_CMD_BOOTMENU) += bootmenu.o
>  obj-$(CONFIG_CMD_BOOTSTAGE) += bootstage.o
> diff --git a/cmd/bootcount.c b/cmd/bootcount.c
> new file mode 100644
> index 0000000..c358418
> --- /dev/null
> +++ b/cmd/bootcount.c
> @@ -0,0 +1,61 @@
> +// SPDX-License-Identifier: GPL-2.0+
> +
> +#include <common.h>
> +#include <command.h>
> +#include <bootcount.h>
> +
> +static int do_bootcount_print(cmd_tbl_t *cmdtp, int flag, int argc,
> +			      char * const argv[])
> +{
> +	printf("%lu\n", bootcount_load());
> +	return CMD_RET_SUCCESS;
> +}
> +
> +static int do_bootcount_reset(cmd_tbl_t *cmdtp, int flag, int argc,
> +			      char * const argv[])
> +{
> +	/*
> +	 * note that we're explicitly not resetting the environment
> +	 * variable, so you still have the old bootcounter available
> +	 */
> +	bootcount_store(0);
> +	return CMD_RET_SUCCESS;
> +}
> +
> +static cmd_tbl_t bootcount_sub[] = {
> +	U_BOOT_CMD_MKENT(print, 1, 1, do_bootcount_print, "", ""),
> +	U_BOOT_CMD_MKENT(reset, 1, 1, do_bootcount_reset, "", ""),
> +};
> +
> +static int do_bootcount(cmd_tbl_t *cmdtp, int flag, int argc,
> +			char * const argv[])
> +{
> +	cmd_tbl_t *cp;
> +
> +	if (argc < 2)
> +		return CMD_RET_USAGE;
> +
> +	/* drop initial "bootcount" arg */
> +	argc--;
> +	argv++;
> +
> +	cp = find_cmd_tbl(argv[0], bootcount_sub,
> ARRAY_SIZE(bootcount_sub));
> +	if (cp)
> +		return cp->cmd(cmdtp, flag, argc, argv);
> +
> +	return CMD_RET_USAGE;
> +}
> +
> +#if CONFIG_IS_ENABLED(SYS_LONGHELP)
> +static char bootcount_help_text[] =
> +	"print - print current bootcounter\n"
> +	"reset - reset the bootcounter"
> +	;
> +#endif
> +
> +U_BOOT_CMD(bootcount, 2, 1, do_bootcount,
> +	   "bootcount",
> +#if CONFIG_IS_ENABLED(SYS_LONGHELP)
> +	   bootcount_help_text
> +#endif
> +);

Reviewed-by: Lukasz Majewski <lukma@denx.de>


Best regards,

Lukasz Majewski

--

DENX Software Engineering GmbH,      Managing Director: Wolfgang Denk
HRB 165235 Munich, Office: Kirchenstr.5, D-82194 Groebenzell, Germany
Phone: (+49)-8142-66989-10 Fax: (+49)-8142-66989-80 Email: wd@denx.de
Tom Rini May 24, 2018, 12:41 p.m. UTC | #2
On Sat, May 12, 2018 at 05:49:47AM +0000, Alex Kiernan wrote:

> Add a command to manipulate the bootcounter. This is useful if you can
> run device recovery from inside U-Boot and need to reset the bootcounter
> after executing that process as part of altbootcmd.
> 
> Signed-off-by: Alex Kiernan <alex.kiernan@gmail.com>
> Reviewed-by: Lukasz Majewski <lukma@denx.de>

Applied to u-boot/master, thanks!
diff mbox series

Patch

diff --git a/cmd/Kconfig b/cmd/Kconfig
index eb14e08..825d4ad 100644
--- a/cmd/Kconfig
+++ b/cmd/Kconfig
@@ -1218,6 +1218,13 @@  config CMD_BMP
 	  the image into RAM, then using this command to look at it or display
 	  it.
 
+config CMD_BOOTCOUNT
+	bool "bootcount"
+	depends on BOOTCOUNT_LIMIT
+	help
+	  Enable the bootcount command, which allows interrogation and
+	  reset of the bootcounter.
+
 config CMD_BSP
 	bool "Enable board-specific commands"
 	help
diff --git a/cmd/Makefile b/cmd/Makefile
index 0d7322e..c05dc2b 100644
--- a/cmd/Makefile
+++ b/cmd/Makefile
@@ -22,6 +22,7 @@  obj-$(CONFIG_CMD_BEDBUG) += bedbug.o
 obj-$(CONFIG_CMD_BINOP) += binop.o
 obj-$(CONFIG_CMD_BLOCK_CACHE) += blkcache.o
 obj-$(CONFIG_CMD_BMP) += bmp.o
+obj-$(CONFIG_CMD_BOOTCOUNT) += bootcount.o
 obj-$(CONFIG_CMD_BOOTEFI) += bootefi.o
 obj-$(CONFIG_CMD_BOOTMENU) += bootmenu.o
 obj-$(CONFIG_CMD_BOOTSTAGE) += bootstage.o
diff --git a/cmd/bootcount.c b/cmd/bootcount.c
new file mode 100644
index 0000000..c358418
--- /dev/null
+++ b/cmd/bootcount.c
@@ -0,0 +1,61 @@ 
+// SPDX-License-Identifier: GPL-2.0+
+
+#include <common.h>
+#include <command.h>
+#include <bootcount.h>
+
+static int do_bootcount_print(cmd_tbl_t *cmdtp, int flag, int argc,
+			      char * const argv[])
+{
+	printf("%lu\n", bootcount_load());
+	return CMD_RET_SUCCESS;
+}
+
+static int do_bootcount_reset(cmd_tbl_t *cmdtp, int flag, int argc,
+			      char * const argv[])
+{
+	/*
+	 * note that we're explicitly not resetting the environment
+	 * variable, so you still have the old bootcounter available
+	 */
+	bootcount_store(0);
+	return CMD_RET_SUCCESS;
+}
+
+static cmd_tbl_t bootcount_sub[] = {
+	U_BOOT_CMD_MKENT(print, 1, 1, do_bootcount_print, "", ""),
+	U_BOOT_CMD_MKENT(reset, 1, 1, do_bootcount_reset, "", ""),
+};
+
+static int do_bootcount(cmd_tbl_t *cmdtp, int flag, int argc,
+			char * const argv[])
+{
+	cmd_tbl_t *cp;
+
+	if (argc < 2)
+		return CMD_RET_USAGE;
+
+	/* drop initial "bootcount" arg */
+	argc--;
+	argv++;
+
+	cp = find_cmd_tbl(argv[0], bootcount_sub, ARRAY_SIZE(bootcount_sub));
+	if (cp)
+		return cp->cmd(cmdtp, flag, argc, argv);
+
+	return CMD_RET_USAGE;
+}
+
+#if CONFIG_IS_ENABLED(SYS_LONGHELP)
+static char bootcount_help_text[] =
+	"print - print current bootcounter\n"
+	"reset - reset the bootcounter"
+	;
+#endif
+
+U_BOOT_CMD(bootcount, 2, 1, do_bootcount,
+	   "bootcount",
+#if CONFIG_IS_ENABLED(SYS_LONGHELP)
+	   bootcount_help_text
+#endif
+);