From fb613e9e67ae52ba07585845f52d751916cf480a Mon Sep 17 00:00:00 2001 From: Geoff Levand Date: Sun, 12 Apr 2009 15:11:58 +0000 Subject: [PATCH] Add ncurses boot option editor Add an ncurses boot option editor. Signed-off-by: Geoff Levand Signed-off-by: Jeremy Kerr --- rules.mk | 3 +- ui/ncurses/nc-ked.c | 348 ++++++++++++++++++++++++++++++++++++++++++++ ui/ncurses/nc-ked.h | 68 +++++++++ 3 files changed, 418 insertions(+), 1 deletion(-) create mode 100644 ui/ncurses/nc-ked.c create mode 100644 ui/ncurses/nc-ked.h diff --git a/rules.mk b/rules.mk index d3dae15..de5f013 100644 --- a/rules.mk +++ b/rules.mk @@ -50,7 +50,8 @@ discover_objs = discover/event.o discover/user-event.o discover/udev.o \ # client objs ui_common_objs = ui/common/discover-client.o ui/common/loader.o \ ui/common/ui-system.o ui/common/url.o -ncurses_objs = ui/ncurses/nc-scr.o ui/ncurses/nc-menu.o +ncurses_objs = ui/ncurses/nc-scr.o ui/ncurses/nc-menu.o \ + ui/ncurses/nc-ked.o twin_objs = ui/twin/pb-twin.o # Makefiles diff --git a/ui/ncurses/nc-ked.c b/ui/ncurses/nc-ked.c new file mode 100644 index 0000000..3bdbd6c --- /dev/null +++ b/ui/ncurses/nc-ked.c @@ -0,0 +1,348 @@ +/* + * Copyright (C) 2009 Sony Computer Entertainment Inc. + * Copyright 2009 Sony Corp. + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; version 2 of the License. + * + * 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. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +#define _GNU_SOURCE + +#include +#include + +#include "log/log.h" +#include "talloc/talloc.h" +#include "nc-ked.h" + +static struct ked *ked_from_scr(struct nc_scr *scr) +{ + struct ked *ked; + + assert(scr->sig == pb_ked_sig); + ked = (struct ked *)((char *)scr - (size_t)&((struct ked *)0)->scr); + assert(ked->scr.sig == pb_ked_sig); + return ked; +} + +static struct ked *ked_from_arg(void *arg) +{ + struct ked *ked = arg; + + assert(ked->scr.sig == pb_ked_sig); + return ked; +} + +/** + * ked_move_cursor - Move the cursor, setting correct attributes. + * @req: An ncurses request or char to send to form_driver(). + */ + +static void ked_move_cursor(struct ked *ked, int req) +{ + wchgat(ked->scr.sub_ncw, 1, ked_attr_field_selected, 0, 0); + form_driver(ked->ncf, req); + wchgat(ked->scr.sub_ncw, 1, ked->attr_cursor, 0, 0); + wrefresh(ked->scr.main_ncw); +} + +/** + * ked_insert_mode_set - Set the insert mode. + */ + +static void ked_insert_mode_set(struct ked *ked, int req) +{ + switch (req) { + case REQ_INS_MODE: + ked->attr_cursor = ked_attr_cursor_ins; + break; + case REQ_OVL_MODE: + ked->attr_cursor = ked_attr_cursor_ovl; + break; + default: + assert(0 && "bad req"); + break; + } + ked_move_cursor(ked, req); +} + +/** + * ked_insert_mode_tog - Toggle the insert mode. + */ + +static void ked_insert_mode_tog(struct ked *ked) +{ + if (ked->attr_cursor == ked_attr_cursor_ins) + ked_insert_mode_set(ked, REQ_OVL_MODE); + else + ked_insert_mode_set(ked, REQ_INS_MODE); +} + +/** + * ked_move_field - Move selected field, setting correct attributes. + * @req: An ncurses request to send to form_driver(). + */ + +static void ked_move_field(struct ked *ked, int req) +{ + set_field_back(current_field(ked->ncf), ked_attr_field_normal); + form_driver(ked->ncf, req); + set_field_back(current_field(ked->ncf), ked_attr_field_selected); + ked_move_cursor(ked, REQ_END_FIELD); +} + +static int ked_post(struct nc_scr *scr) +{ + struct ked *ked = ked_from_scr(scr); + + post_form(ked->ncf); + + nc_scr_frame_draw(scr); + ked_move_field(ked, REQ_FIRST_FIELD); + ked_move_field(ked, REQ_END_FIELD); + ked_insert_mode_set(ked, REQ_INS_MODE); + + redrawwin(ked->scr.main_ncw); + wrefresh(ked->scr.main_ncw); + + return 0; +} + +static int ked_unpost(struct nc_scr *scr) +{ + return unpost_form(ked_from_scr(scr)->ncf); +} + +static void ked_resize(struct nc_scr *scr) +{ + /* FIXME: forms can't be resized, need to recreate here */ + ked_unpost(scr); + ked_post(scr); +} + +/** + * ked_chomp - Eat leading and trailing WS. + */ + +static char *ked_chomp(char *s) +{ + char *start; + char *end; + char *const s_end = s + strlen(s); + + for (; s < s_end; s++) + if (*s != ' ' && *s != '\t') + break; + start = s; + + for (++s; s < s_end; s++) + if (*s != ' ' && *s != '\t') + end = s; + *(end + 1) = 0; + return start; +} + +static struct pb_kexec_data *ked_prepare_data(struct ked *ked) +{ + struct pb_kexec_data *kd; + char *s; + + kd = talloc(ked, struct pb_kexec_data); + + if (!kd) + return NULL; + + s = ked_chomp(field_buffer(ked->fields[0], 0)); + kd->image = *s ? talloc_strdup(kd, s) : NULL; + + s = ked_chomp(field_buffer(ked->fields[1], 0)); + kd->initrd = *s ? talloc_strdup(kd, s) : NULL; + + s = ked_chomp(field_buffer(ked->fields[2], 0)); + kd->args = *s ? talloc_strdup(kd, s) : NULL; + + return kd; +} + +/** + * ked_process_key - Process a user keystroke. + * + * Called from the cui via the scr:process_key method. + */ + +static void ked_process_key(struct nc_scr *scr) +{ + struct ked *ked = ked_from_scr(scr); + + while (1) { + int c = getch(); + + switch (c) { + default: + ked_move_cursor(ked, c); + break; + case ERR: + return; + + /* hot keys */ + case 2: { /* CTRL-B */ + struct pb_kexec_data *kd; + + form_driver(ked->ncf, REQ_VALIDATION); + kd = ked_prepare_data(ked); + ked->on_exit(ked, ked_boot, kd); + nc_flush_keys(); + return; + } + case 27: /* ESC */ + ked->on_exit(ked, ked_cancel, NULL); + nc_flush_keys(); + return; + case '\n': + case '\r': { + struct pb_kexec_data *kd; + + form_driver(ked->ncf, REQ_VALIDATION); + kd = ked_prepare_data(ked); + ked->on_exit(ked, ked_update, kd); + nc_flush_keys(); + return; + } + + /* insert mode */ + case KEY_IC: + ked_insert_mode_tog(ked); + break; + + /* form nav */ + case KEY_PPAGE: + ked_move_field(ked, REQ_FIRST_FIELD); + break; + case KEY_NPAGE: + ked_move_field(ked, REQ_LAST_FIELD); + break; + case KEY_DOWN: + ked_move_field(ked, REQ_NEXT_FIELD); + break; + case KEY_UP: + ked_move_field(ked, REQ_PREV_FIELD); + break; + + /* field nav */ + case KEY_HOME: + ked_move_cursor(ked, REQ_BEG_FIELD); + break; + case KEY_END: + ked_move_cursor(ked, REQ_END_FIELD); + break; + case KEY_LEFT: + ked_move_cursor(ked, REQ_LEFT_CHAR); + break; + case KEY_RIGHT: + ked_move_cursor(ked, REQ_RIGHT_CHAR); + break; + case KEY_BACKSPACE: + ked_move_cursor(ked, REQ_LEFT_CHAR); + ked_move_cursor(ked, REQ_DEL_CHAR); + break; + case KEY_DC: + ked_move_cursor(ked, REQ_DEL_CHAR); + break; + } + } +} + +/** + * ked_destructor - The talloc destructor for a ked. + */ + +static int ked_destructor(void *arg) +{ + struct ked *ked = ked_from_arg(arg); + FIELD **f; + + for (f = ked->fields; *f; f++) + free_field(*f); + + free_form(ked->ncf); + ked->scr.sig = pb_removed_sig; + + return 0; +} + +static FIELD *ked_setup_field(unsigned int y, unsigned int x, char *str) +{ + FIELD *f; + + f = new_field(1, COLS - 1 - x, y, x, 0, 0); + field_opts_off(f, O_STATIC | O_WRAP); + set_max_field(f, 256); + set_field_buffer(f, 0, str); + set_field_status(f, 0); + return f; +} + +static FIELD *ked_setup_label(unsigned int y, unsigned int x, char *str) +{ + FIELD *f; + + f = new_field(1, strlen(str), y, x, 0, 0); + field_opts_off(f, O_ACTIVE); + set_field_buffer(f, 0, str); + return f; +} + +struct ked *ked_init(void *ui_ctx, const struct pb_kexec_data *kd, + void (*on_exit)(struct ked *, enum ked_result, struct pb_kexec_data *)) +{ + struct ked *ked; + + pb_log("%s: image: '%s'\n", __func__, kd->image); + pb_log("%s: initrd: '%s'\n", __func__, kd->initrd); + pb_log("%s: args: '%s'\n", __func__, kd->args); + + assert(on_exit); + + ked = talloc_zero(ui_ctx, struct ked); + + if (!ked) + return NULL; + + talloc_set_destructor(ked, ked_destructor); + + nc_scr_init(&ked->scr, pb_ked_sig, 0, ui_ctx, ked_process_key, + ked_post, ked_unpost, ked_resize); + + ked->scr.frame.title = talloc_strdup(ked, "Petitboot Option Editor"); + ked->scr.frame.help = talloc_strdup(ked, + "ESC=cancel, Enter=accept, Ctrl-b=boot"); + + ked->on_exit = on_exit; + + ked->fields = talloc_array(ked, FIELD *, 7); + + ked->fields[0] = ked_setup_field(0, 9, kd->image); + ked->fields[1] = ked_setup_field(1, 9, kd->initrd); + ked->fields[2] = ked_setup_field(2, 9, kd->args); + ked->fields[3] = ked_setup_label(0, 1, "image:"); + ked->fields[4] = ked_setup_label(1, 1, "initrd:"); + ked->fields[5] = ked_setup_label(2, 1, "args:"); + ked->fields[6] = NULL; + + ked->ncf = new_form(ked->fields); + + set_form_win(ked->ncf, ked->scr.main_ncw); + set_form_sub(ked->ncf, ked->scr.sub_ncw); + + return ked; +} diff --git a/ui/ncurses/nc-ked.h b/ui/ncurses/nc-ked.h new file mode 100644 index 0000000..36ed4f1 --- /dev/null +++ b/ui/ncurses/nc-ked.h @@ -0,0 +1,68 @@ +/* + * Copyright (C) 2009 Sony Computer Entertainment Inc. + * Copyright 2009 Sony Corp. + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; version 2 of the License. + * + * 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. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +#if !defined(_PB_NC_KED_H) +#define _PB_NC_KED_H + +#include +#include + +#include "pb-protocol/pb-protocol.h" +#include "ui/common/ui-system.h" +#include "nc-scr.h" + +enum ked_attr_field { + ked_attr_field_normal = A_NORMAL, + ked_attr_field_selected = A_REVERSE, +}; + +enum ked_attr_cursor { + ked_attr_cursor_ins = A_NORMAL, + ked_attr_cursor_ovl = A_NORMAL | A_UNDERLINE, +}; + +/** + * enum ked_result - Result code for ked:on_exit(). + * @ked_cancel: The user canceled the operation. + * @ked_update: The args were updated. + * @ked_boot: The user requested a boot of this item. + */ + +enum ked_result { + ked_cancel, + ked_update, + ked_boot, +}; + +/** + * struct ked - kexec args editor. + */ + +struct ked { + struct nc_scr scr; + FORM *ncf; + FIELD **fields; + enum ked_attr_cursor attr_cursor; + void (*on_exit)(struct ked *ked, enum ked_result result, + struct pb_kexec_data *kd); +}; + +struct ked *ked_init(void *ui_ctx, const struct pb_kexec_data *kd, + void (*on_exit)(struct ked *, enum ked_result, struct pb_kexec_data *)); + +#endif -- 2.39.2