123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288 |
- // SPDX-License-Identifier: GPL-2.0+
- /*
- * Provide a menu of available bootflows and related options
- *
- * Copyright 2022 Google LLC
- * Written by Simon Glass <sjg@chromium.org>
- */
- #define LOG_CATEGORY UCLASS_BOOTSTD
- #include <common.h>
- #include <bootflow.h>
- #include <bootstd.h>
- #include <cli.h>
- #include <dm.h>
- #include <expo.h>
- #include <malloc.h>
- #include <menu.h>
- #include <video_console.h>
- #include <watchdog.h>
- #include <linux/delay.h>
- #include "bootflow_internal.h"
- /**
- * struct menu_priv - information about the menu
- *
- * @num_bootflows: Number of bootflows in the menu
- */
- struct menu_priv {
- int num_bootflows;
- };
- int bootflow_menu_new(struct expo **expp)
- {
- struct udevice *last_bootdev;
- struct scene_obj_menu *menu;
- struct menu_priv *priv;
- struct bootflow *bflow;
- struct scene *scn;
- struct expo *exp;
- void *logo;
- int ret, i;
- priv = calloc(1, sizeof(*priv));
- if (!priv)
- return log_msg_ret("prv", -ENOMEM);
- ret = expo_new("bootflows", priv, &exp);
- if (ret)
- return log_msg_ret("exp", ret);
- ret = scene_new(exp, "main", MAIN, &scn);
- if (ret < 0)
- return log_msg_ret("scn", ret);
- ret |= scene_txt_str(scn, "prompt", OBJ_PROMPT, STR_PROMPT,
- "UP and DOWN to choose, ENTER to select", NULL);
- ret = scene_menu(scn, "main", OBJ_MENU, &menu);
- ret |= scene_obj_set_pos(scn, OBJ_MENU, MARGIN_LEFT, 100);
- ret |= scene_txt_str(scn, "title", OBJ_MENU_TITLE, STR_MENU_TITLE,
- "U-Boot - Boot Menu", NULL);
- ret |= scene_menu_set_title(scn, OBJ_MENU, OBJ_PROMPT);
- logo = video_get_u_boot_logo();
- if (logo) {
- ret |= scene_img(scn, "ulogo", OBJ_U_BOOT_LOGO, logo, NULL);
- ret |= scene_obj_set_pos(scn, OBJ_U_BOOT_LOGO, -4, 4);
- }
- ret |= scene_txt_str(scn, "cur_item", OBJ_POINTER, STR_POINTER, ">",
- NULL);
- ret |= scene_menu_set_pointer(scn, OBJ_MENU, OBJ_POINTER);
- if (ret < 0)
- return log_msg_ret("new", -EINVAL);
- last_bootdev = NULL;
- for (ret = bootflow_first_glob(&bflow), i = 0; !ret && i < 36;
- ret = bootflow_next_glob(&bflow), i++) {
- char str[2], *label, *key;
- uint preview_id;
- bool add_gap;
- if (bflow->state != BOOTFLOWST_READY)
- continue;
- *str = i < 10 ? '0' + i : 'A' + i - 10;
- str[1] = '\0';
- key = strdup(str);
- if (!key)
- return log_msg_ret("key", -ENOMEM);
- label = strdup(dev_get_parent(bflow->dev)->name);
- if (!label) {
- free(key);
- return log_msg_ret("nam", -ENOMEM);
- }
- add_gap = last_bootdev != bflow->dev;
- last_bootdev = bflow->dev;
- ret = expo_str(exp, "prompt", STR_POINTER, ">");
- ret |= scene_txt_str(scn, "label", ITEM_LABEL + i,
- STR_LABEL + i, label, NULL);
- ret |= scene_txt_str(scn, "desc", ITEM_DESC + i, STR_DESC + i,
- bflow->os_name ? bflow->os_name :
- bflow->name, NULL);
- ret |= scene_txt_str(scn, "key", ITEM_KEY + i, STR_KEY + i, key,
- NULL);
- preview_id = 0;
- if (bflow->logo) {
- preview_id = ITEM_PREVIEW + i;
- ret |= scene_img(scn, "preview", preview_id,
- bflow->logo, NULL);
- }
- ret |= scene_menuitem(scn, OBJ_MENU, "item", ITEM + i,
- ITEM_KEY + i, ITEM_LABEL + i,
- ITEM_DESC + i, preview_id,
- add_gap ? SCENEMIF_GAP_BEFORE : 0,
- NULL);
- if (ret < 0)
- return log_msg_ret("itm", -EINVAL);
- ret = 0;
- priv->num_bootflows++;
- }
- ret = scene_arrange(scn);
- if (ret)
- return log_msg_ret("arr", ret);
- *expp = exp;
- return 0;
- }
- int bootflow_menu_apply_theme(struct expo *exp, ofnode node)
- {
- struct menu_priv *priv = exp->priv;
- struct scene *scn;
- u32 font_size;
- int ret;
- log_debug("Applying theme %s\n", ofnode_get_name(node));
- scn = expo_lookup_scene_id(exp, MAIN);
- if (!scn)
- return log_msg_ret("scn", -ENOENT);
- /* Avoid error-checking optional items */
- if (!ofnode_read_u32(node, "font-size", &font_size)) {
- int i;
- log_debug("font size %d\n", font_size);
- scene_txt_set_font(scn, OBJ_PROMPT, NULL, font_size);
- scene_txt_set_font(scn, OBJ_POINTER, NULL, font_size);
- for (i = 0; i < priv->num_bootflows; i++) {
- ret = scene_txt_set_font(scn, ITEM_DESC + i, NULL,
- font_size);
- if (ret)
- return log_msg_ret("des", ret);
- scene_txt_set_font(scn, ITEM_KEY + i, NULL, font_size);
- scene_txt_set_font(scn, ITEM_LABEL + i, NULL,
- font_size);
- }
- }
- ret = scene_arrange(scn);
- if (ret)
- return log_msg_ret("arr", ret);
- return 0;
- }
- int bootflow_menu_run(struct bootstd_priv *std, bool text_mode,
- struct bootflow **bflowp)
- {
- struct cli_ch_state s_cch, *cch = &s_cch;
- struct bootflow *sel_bflow;
- struct udevice *dev;
- struct expo *exp;
- uint sel_id;
- bool done;
- int ret;
- cli_ch_init(cch);
- sel_bflow = NULL;
- *bflowp = NULL;
- ret = bootflow_menu_new(&exp);
- if (ret)
- return log_msg_ret("exp", ret);
- if (ofnode_valid(std->theme)) {
- ret = bootflow_menu_apply_theme(exp, std->theme);
- if (ret)
- return log_msg_ret("thm", ret);
- }
- /* For now we only support a video console */
- ret = uclass_first_device_err(UCLASS_VIDEO, &dev);
- if (ret)
- return log_msg_ret("vid", ret);
- ret = expo_set_display(exp, dev);
- if (ret)
- return log_msg_ret("dis", ret);
- ret = expo_set_scene_id(exp, MAIN);
- if (ret)
- return log_msg_ret("scn", ret);
- if (text_mode)
- expo_set_text_mode(exp, text_mode);
- done = false;
- do {
- struct expo_action act;
- int ichar, key;
- ret = expo_render(exp);
- if (ret)
- break;
- ichar = cli_ch_process(cch, 0);
- if (!ichar) {
- while (!ichar && !tstc()) {
- schedule();
- mdelay(2);
- ichar = cli_ch_process(cch, -ETIMEDOUT);
- }
- if (!ichar) {
- ichar = getchar();
- ichar = cli_ch_process(cch, ichar);
- }
- }
- key = 0;
- if (ichar) {
- key = bootmenu_conv_key(ichar);
- if (key == BKEY_NONE)
- key = ichar;
- }
- if (!key)
- continue;
- ret = expo_send_key(exp, key);
- if (ret)
- break;
- ret = expo_action_get(exp, &act);
- if (!ret) {
- switch (act.type) {
- case EXPOACT_SELECT:
- sel_id = act.select.id;
- done = true;
- break;
- case EXPOACT_QUIT:
- done = true;
- break;
- default:
- break;
- }
- }
- } while (!done);
- if (ret)
- return log_msg_ret("end", ret);
- if (sel_id) {
- struct bootflow *bflow;
- int i;
- for (ret = bootflow_first_glob(&bflow), i = 0; !ret && i < 36;
- ret = bootflow_next_glob(&bflow), i++) {
- if (i == sel_id - ITEM) {
- sel_bflow = bflow;
- break;
- }
- }
- }
- expo_destroy(exp);
- if (!sel_bflow)
- return -EAGAIN;
- *bflowp = sel_bflow;
- return 0;
- }
|