boot -m -t 3 already opens a boot menu with a countdown of 3 seconds
before selecting the first element. So far, only way to influence
preselection was shifting around boot entries, so they are iterated over
differently. Add a new -M option that works analogously to -m, but
takes an integer index of the boot menu entry to preselect. This allows
simple customizable interactive boots:

    #!/bin/sh
    boot -M "$nv.bootmenu_default" -t 3 mmc0.0

With mmc0.0 containing multiple bootloader spec files that would be iterated
over in lexical order. The index is 1-based like the index displayed in
the boot menu.

Signed-off-by: Ahmad Fatoum <[email protected]>
---
For lexically ordered bootspec files, following series is required:
https://lore.barebox.org/barebox/[email protected]/T/#t
---
 commands/boot.c | 18 +++++++++++++++---
 common/boot.c   | 10 +++++++++-
 include/boot.h  |  2 +-
 3 files changed, 25 insertions(+), 5 deletions(-)

diff --git a/commands/boot.c b/commands/boot.c
index 18f4e36ec733..485559bc46b8 100644
--- a/commands/boot.c
+++ b/commands/boot.c
@@ -28,13 +28,14 @@ static int do_boot(int argc, char *argv[])
        char *freep = NULL;
        int opt, ret = 0, do_list = 0, do_menu = 0;
        int dryrun = 0, verbose = 0, timeout = -1;
+       unsigned default_menu_entry = 0;
        struct bootentries *entries;
        struct bootentry *entry;
        void *handle;
        const char *name;
        char *(*next)(void *);
 
-       while ((opt = getopt(argc, argv, "vldmt:w:")) > 0) {
+       while ((opt = getopt(argc, argv, "vldmM:t:w:")) > 0) {
                switch (opt) {
                case 'v':
                        verbose++;
@@ -45,6 +46,16 @@ static int do_boot(int argc, char *argv[])
                case 'd':
                        dryrun = 1;
                        break;
+               case 'M':
+                       /* To simplify scripting, an empty string is treated as 
1 */
+                       if (*optarg == '\0') {
+                               default_menu_entry = 1;
+                       } else {
+                               ret = kstrtouint(optarg, 0, 
&default_menu_entry);
+                               if (ret)
+                                       return ret;
+                       }
+                       fallthrough;
                case 'm':
                        do_menu = 1;
                        break;
@@ -104,7 +115,7 @@ static int do_boot(int argc, char *argv[])
        if (do_list)
                bootsources_list(entries);
        else if (do_menu)
-               bootsources_menu(entries, timeout);
+               bootsources_menu(entries, default_menu_entry, timeout);
 
        ret = 0;
 out:
@@ -136,6 +147,7 @@ BAREBOX_CMD_HELP_OPT ("-v","Increase verbosity")
 BAREBOX_CMD_HELP_OPT ("-d","Dryrun. See what happens but do no actually boot")
 BAREBOX_CMD_HELP_OPT ("-l","List available boot sources")
 BAREBOX_CMD_HELP_OPT ("-m","Show a menu with boot options")
+BAREBOX_CMD_HELP_OPT ("-M INDEX","Show a menu with boot options with entry 
INDEX preselected")
 BAREBOX_CMD_HELP_OPT ("-w SECS","Start watchdog with timeout SECS before 
booting")
 BAREBOX_CMD_HELP_OPT ("-t SECS","specify timeout in SECS")
 BAREBOX_CMD_HELP_END
@@ -143,7 +155,7 @@ BAREBOX_CMD_HELP_END
 BAREBOX_CMD_START(boot)
        .cmd    = do_boot,
        BAREBOX_CMD_DESC("boot from script, device, ...")
-       BAREBOX_CMD_OPTS("[-vdlmwt] [BOOTSRC...]")
+       BAREBOX_CMD_OPTS("[-vdlmMwt] [BOOTSRC...]")
        BAREBOX_CMD_GROUP(CMD_GRP_BOOT)
        BAREBOX_CMD_HELP(cmd_boot_help)
 BAREBOX_CMD_END
diff --git a/common/boot.c b/common/boot.c
index 8220b8d3fbd7..52b03eb64d2e 100644
--- a/common/boot.c
+++ b/common/boot.c
@@ -313,10 +313,12 @@ int bootentry_create_from_name(struct bootentries 
*bootentries,
 /*
  * bootsources_menu - show a menu from an array of names
  */
-void bootsources_menu(struct bootentries *bootentries, int timeout)
+void bootsources_menu(struct bootentries *bootentries,
+                     unsigned default_entry, int timeout)
 {
        struct bootentry *entry;
        struct menu_entry *back_entry;
+       int i = 1;
 
        if (!IS_ENABLED(CONFIG_MENU)) {
                pr_warn("no menu support available\n");
@@ -328,6 +330,9 @@ void bootsources_menu(struct bootentries *bootentries, int 
timeout)
                        entry->me.display = xstrdup(entry->title);
                entry->me.action = bootsource_action;
                menu_add_entry(bootentries->menu, &entry->me);
+
+               if (i++ == default_entry)
+                       bootentries->menu->selected = &entry->me;
        }
 
        back_entry = xzalloc(sizeof(*back_entry));
@@ -336,6 +341,9 @@ void bootsources_menu(struct bootentries *bootentries, int 
timeout)
        back_entry->non_re_ent = 1;
        menu_add_entry(bootentries->menu, back_entry);
 
+       if (i == default_entry)
+               bootentries->menu->selected = back_entry;
+
        if (timeout >= 0)
                bootentries->menu->auto_select = timeout;
 
diff --git a/include/boot.h b/include/boot.h
index 3d5dd1cb6e2c..c9e8c0fd0d87 100644
--- a/include/boot.h
+++ b/include/boot.h
@@ -50,7 +50,7 @@ struct bootentries *bootentries_alloc(void);
 void bootentries_free(struct bootentries *bootentries);
 int bootentry_create_from_name(struct bootentries *bootentries,
                                      const char *name);
-void bootsources_menu(struct bootentries *bootentries, int timeout);
+void bootsources_menu(struct bootentries *bootentries, unsigned default_entry, 
int timeout);
 void bootsources_list(struct bootentries *bootentries);
 int boot_entry(struct bootentry *be, int verbose, int dryrun);
 
-- 
2.34.1


Reply via email to