git.ozlabs.org Git - petitboot/blob - ui/ncurses/nc-ked.c

   1 /*
   2  *  Copyright (C) 2009 Sony Computer Entertainment Inc.
   3  *  Copyright 2009 Sony Corp.
   4  *
   5  *  This program is free software; you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation; version 2 of the License.
   8  *
   9  *  This program is distributed in the hope that it will be useful,
  10  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  *  GNU General Public License for more details.
  13  *
  14  *  You should have received a copy of the GNU General Public License
  15  *  along with this program; if not, write to the Free Software
  16  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17  */
  18
  19 #define _GNU_SOURCE
  20
  21 #include <assert.h>
  22 #include <string.h>
  23
  24 #include "log/log.h"
  25 #include "talloc/talloc.h"
  26 #include "nc-ked.h"
  27
  28 static struct ked *ked_from_scr(struct nc_scr *scr)
  29 {
  30         struct ked *ked;
  31
  32         assert(scr->sig == pb_ked_sig);
  33         ked = (struct ked *)((char *)scr - (size_t)&((struct ked *)0)->scr);
  34         assert(ked->scr.sig == pb_ked_sig);
  35         return ked;
  36 }
  37
  38 static struct ked *ked_from_arg(void *arg)
  39 {
  40         struct ked *ked = arg;
  41
  42         assert(ked->scr.sig == pb_ked_sig);
  43         return ked;
  44 }
  45
  46 /**
  47  * ked_move_cursor - Move the cursor, setting correct attributes.
  48  * @req: An ncurses request or char to send to form_driver().
  49  */
  50
  51 static int ked_move_cursor(struct ked *ked, int req)
  52 {
  53         int result;
  54
  55         wchgat(ked->scr.sub_ncw, 1, ked_attr_field_selected, 0, 0);
  56         result = form_driver(ked->ncf, req);
  57         wchgat(ked->scr.sub_ncw, 1, ked->attr_cursor, 0, 0);
  58         wrefresh(ked->scr.main_ncw);
  59         return result;
  60 }
  61
  62 /**
  63  * ked_insert_mode_set - Set the insert mode.
  64  */
  65
  66 static void ked_insert_mode_set(struct ked *ked, int req)
  67 {
  68         switch (req) {
  69         case REQ_INS_MODE:
  70                 ked->attr_cursor = ked_attr_cursor_ins;
  71                 break;
  72         case REQ_OVL_MODE:
  73                 ked->attr_cursor = ked_attr_cursor_ovl;
  74                 break;
  75         default:
  76                 assert(0 && "bad req");
  77                 break;
  78         }
  79         ked_move_cursor(ked, req);
  80 }
  81
  82 /**
  83  * ked_insert_mode_tog - Toggle the insert mode.
  84  */
  85
  86 static void ked_insert_mode_tog(struct ked *ked)
  87 {
  88         if (ked->attr_cursor == ked_attr_cursor_ins)
  89                 ked_insert_mode_set(ked, REQ_OVL_MODE);
  90         else
  91                 ked_insert_mode_set(ked, REQ_INS_MODE);
  92 }
  93
  94 /**
  95  * ked_move_field - Move selected field, setting correct attributes.
  96  * @req: An ncurses request to send to form_driver().
  97  */
  98
  99 static int ked_move_field(struct ked *ked, int req)
 100 {
 101         int result;
 102
 103         set_field_back(current_field(ked->ncf), ked_attr_field_normal);
 104         result = form_driver(ked->ncf, req);
 105         set_field_back(current_field(ked->ncf), ked_attr_field_selected);
 106         ked_move_cursor(ked, REQ_END_FIELD);
 107         return result;
 108 }
 109
 110 static int ked_post(struct nc_scr *scr)
 111 {
 112         struct ked *ked = ked_from_scr(scr);
 113
 114         post_form(ked->ncf);
 115
 116         nc_scr_frame_draw(scr);
 117         ked_move_field(ked, REQ_FIRST_FIELD);
 118         ked_move_field(ked, REQ_END_FIELD);
 119         ked_insert_mode_set(ked, REQ_INS_MODE);
 120
 121         redrawwin(ked->scr.main_ncw);
 122         wrefresh(ked->scr.main_ncw);
 123
 124         return 0;
 125 }
 126
 127 static int ked_unpost(struct nc_scr *scr)
 128 {
 129         return unpost_form(ked_from_scr(scr)->ncf);
 130 }
 131
 132 static void ked_resize(struct nc_scr *scr)
 133 {
 134         /* FIXME: forms can't be resized, need to recreate here */
 135         ked_unpost(scr);
 136         ked_post(scr);
 137 }
 138
 139 /**
 140  * ked_chomp - Eat leading and trailing WS.
 141  */
 142
 143 static char *ked_chomp(char *s)
 144 {
 145         char *start;
 146         char *end;
 147         char *const s_end = s + strlen(s);
 148
 149         for (; s < s_end; s++)
 150                 if (*s != ' ' && *s != '\t')
 151                         break;
 152         start = s;
 153
 154         for (++s; s < s_end; s++)
 155                 if (*s != ' ' && *s != '\t')
 156                         end = s;
 157         *(end + 1) = 0;
 158         return start;
 159 }
 160
 161 static struct pb_kexec_data *ked_prepare_data(struct ked *ked)
 162 {
 163         struct pb_kexec_data *kd;
 164         char *s;
 165
 166         kd = talloc(ked, struct pb_kexec_data);
 167
 168         if (!kd)
 169                 return NULL;
 170
 171         s = ked_chomp(field_buffer(ked->fields[0], 0));
 172         kd->image = *s ? talloc_strdup(kd, s) : NULL;
 173
 174         s = ked_chomp(field_buffer(ked->fields[1], 0));
 175         kd->initrd = *s ? talloc_strdup(kd, s) : NULL;
 176
 177         s = ked_chomp(field_buffer(ked->fields[2], 0));
 178         kd->args = *s ? talloc_strdup(kd, s) : NULL;
 179
 180         return kd;
 181 }
 182
 183 /**
 184  * ked_process_key - Process a user keystroke.
 185  *
 186  * Called from the cui via the scr:process_key method.
 187  */
 188
 189 static void ked_process_key(struct nc_scr *scr)
 190 {
 191         struct ked *ked = ked_from_scr(scr);
 192
 193         while (1) {
 194                 int c = getch();
 195
 196                 if (c == ERR)
 197                         return;
 198
 199                 /* DBGS("%d (%o)\n", c, c); */
 200
 201                 switch (c) {
 202                 default:
 203                         ked_move_cursor(ked, c);
 204                         break;
 205
 206                 /* hot keys */
 207                 case 2: { /* CTRL-B */
 208                         struct pb_kexec_data *kd;
 209
 210                         form_driver(ked->ncf, REQ_VALIDATION);
 211                         kd = ked_prepare_data(ked);
 212                         ked->on_exit(ked, ked_boot, kd);
 213                         nc_flush_keys();
 214                         return;
 215                 }
 216                 case 27: /* ESC */
 217                         ked->on_exit(ked, ked_cancel, NULL);
 218                         nc_flush_keys();
 219                         return;
 220                 case '\n':
 221                 case '\r': {
 222                         struct pb_kexec_data *kd;
 223
 224                         form_driver(ked->ncf, REQ_VALIDATION);
 225                         kd = ked_prepare_data(ked);
 226                         ked->on_exit(ked, ked_update, kd);
 227                         nc_flush_keys();
 228                         return;
 229                 }
 230
 231                 /* insert mode */
 232                 case KEY_IC:
 233                         ked_insert_mode_tog(ked);
 234                         break;
 235
 236                 /* form nav */
 237                 case KEY_PPAGE:
 238                         ked_move_field(ked, REQ_FIRST_FIELD);
 239                         break;
 240                 case KEY_NPAGE:
 241                         ked_move_field(ked, REQ_LAST_FIELD);
 242                         break;
 243                 case KEY_DOWN:
 244                         ked_move_field(ked, REQ_NEXT_FIELD);
 245                         break;
 246                 case KEY_UP:
 247                         ked_move_field(ked, REQ_PREV_FIELD);
 248                         break;
 249
 250                 /* field nav */
 251                 case KEY_HOME:
 252                         ked_move_cursor(ked, REQ_BEG_FIELD);
 253                         break;
 254                 case KEY_END:
 255                         ked_move_cursor(ked, REQ_END_FIELD);
 256                         break;
 257                 case KEY_LEFT:
 258                         ked_move_cursor(ked, REQ_LEFT_CHAR);
 259                         break;
 260                 case KEY_RIGHT:
 261                         ked_move_cursor(ked, REQ_RIGHT_CHAR);
 262                         break;
 263                 case KEY_BACKSPACE:
 264                         if (ked_move_cursor(ked, REQ_LEFT_CHAR) == E_OK)
 265                                 ked_move_cursor(ked, REQ_DEL_CHAR);
 266                         break;
 267                 case KEY_DC:
 268                         ked_move_cursor(ked, REQ_DEL_CHAR);
 269                         break;
 270                 }
 271         }
 272 }
 273
 274 /**
 275  * ked_destructor - The talloc destructor for a ked.
 276  */
 277
 278 static int ked_destructor(void *arg)
 279 {
 280         struct ked *ked = ked_from_arg(arg);
 281         FIELD **f;
 282
 283         for (f = ked->fields; *f; f++)
 284                 free_field(*f);
 285
 286         free_form(ked->ncf);
 287         ked->scr.sig = pb_removed_sig;
 288
 289         return 0;
 290 }
 291
 292 static FIELD *ked_setup_field(unsigned int y, unsigned int x, char *str)
 293 {
 294         FIELD *f;
 295
 296         f = new_field(1, COLS - 1 - x, y, x, 0, 0);
 297         field_opts_off(f, O_STATIC | O_WRAP);
 298         set_max_field(f, 256);
 299         set_field_buffer(f, 0, str);
 300         set_field_status(f, 0);
 301         return f;
 302 }
 303
 304 static FIELD *ked_setup_label(unsigned int y, unsigned int x, char *str)
 305 {
 306         FIELD *f;
 307
 308         f = new_field(1, strlen(str), y, x, 0, 0);
 309         field_opts_off(f, O_ACTIVE);
 310         set_field_buffer(f, 0, str);
 311         return f;
 312 }
 313
 314 struct ked *ked_init(void *ui_ctx, const struct pb_kexec_data *kd,
 315         void (*on_exit)(struct ked *, enum ked_result, struct pb_kexec_data *))
 316 {
 317         struct ked *ked;
 318
 319         pb_log("%s: image:  '%s'\n", __func__, kd->image);
 320         pb_log("%s: initrd: '%s'\n", __func__, kd->initrd);
 321         pb_log("%s: args:   '%s'\n", __func__, kd->args);
 322
 323         assert(on_exit);
 324
 325         ked = talloc_zero(ui_ctx, struct ked);
 326
 327         if (!ked)
 328                 return NULL;
 329
 330         talloc_set_destructor(ked, ked_destructor);
 331
 332         nc_scr_init(&ked->scr, pb_ked_sig, 0, ui_ctx, ked_process_key,
 333                 ked_post, ked_unpost, ked_resize);
 334
 335         ked->scr.frame.title = talloc_strdup(ked, "Petitboot Option Editor");
 336         ked->scr.frame.help = talloc_strdup(ked,
 337                 "ESC=cancel, Enter=accept, Ctrl-b=boot");
 338
 339         ked->on_exit = on_exit;
 340
 341         ked->fields = talloc_array(ked, FIELD *, 7);
 342
 343         ked->fields[0] = ked_setup_field(0, 9, kd->image);
 344         ked->fields[1] = ked_setup_field(1, 9, kd->initrd);
 345         ked->fields[2] = ked_setup_field(2, 9, kd->args);
 346         ked->fields[3] = ked_setup_label(0, 1, "image:");
 347         ked->fields[4] = ked_setup_label(1, 1, "initrd:");
 348         ked->fields[5] = ked_setup_label(2, 1, "args:");
 349         ked->fields[6] = NULL;
 350
 351         ked->ncf = new_form(ked->fields);
 352
 353         set_form_win(ked->ncf, ked->scr.main_ncw);
 354         set_form_sub(ked->ncf, ked->scr.sub_ncw);
 355
 356         return ked;
 357 }