2 * Copyright (C) 2009 Sony Computer Entertainment Inc.
3 * Copyright 2009 Sony Corp.
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.
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.
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
25 #include "talloc/talloc.h"
28 static struct ked *ked_from_scr(struct nc_scr *scr)
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);
38 static struct ked *ked_from_arg(void *arg)
40 struct ked *ked = arg;
42 assert(ked->scr.sig == pb_ked_sig);
47 * ked_move_cursor - Move the cursor, setting correct attributes.
48 * @req: An ncurses request or char to send to form_driver().
51 static void ked_move_cursor(struct ked *ked, int req)
53 wchgat(ked->scr.sub_ncw, 1, ked_attr_field_selected, 0, 0);
54 form_driver(ked->ncf, req);
55 wchgat(ked->scr.sub_ncw, 1, ked->attr_cursor, 0, 0);
56 wrefresh(ked->scr.main_ncw);
60 * ked_insert_mode_set - Set the insert mode.
63 static void ked_insert_mode_set(struct ked *ked, int req)
67 ked->attr_cursor = ked_attr_cursor_ins;
70 ked->attr_cursor = ked_attr_cursor_ovl;
73 assert(0 && "bad req");
76 ked_move_cursor(ked, req);
80 * ked_insert_mode_tog - Toggle the insert mode.
83 static void ked_insert_mode_tog(struct ked *ked)
85 if (ked->attr_cursor == ked_attr_cursor_ins)
86 ked_insert_mode_set(ked, REQ_OVL_MODE);
88 ked_insert_mode_set(ked, REQ_INS_MODE);
92 * ked_move_field - Move selected field, setting correct attributes.
93 * @req: An ncurses request to send to form_driver().
96 static void ked_move_field(struct ked *ked, int req)
98 set_field_back(current_field(ked->ncf), ked_attr_field_normal);
99 form_driver(ked->ncf, req);
100 set_field_back(current_field(ked->ncf), ked_attr_field_selected);
101 ked_move_cursor(ked, REQ_END_FIELD);
104 static int ked_post(struct nc_scr *scr)
106 struct ked *ked = ked_from_scr(scr);
110 nc_scr_frame_draw(scr);
111 ked_move_field(ked, REQ_FIRST_FIELD);
112 ked_move_field(ked, REQ_END_FIELD);
113 ked_insert_mode_set(ked, REQ_INS_MODE);
115 redrawwin(ked->scr.main_ncw);
116 wrefresh(ked->scr.main_ncw);
121 static int ked_unpost(struct nc_scr *scr)
123 return unpost_form(ked_from_scr(scr)->ncf);
126 static void ked_resize(struct nc_scr *scr)
128 /* FIXME: forms can't be resized, need to recreate here */
134 * ked_chomp - Eat leading and trailing WS.
137 static char *ked_chomp(char *s)
141 char *const s_end = s + strlen(s);
143 for (; s < s_end; s++)
144 if (*s != ' ' && *s != '\t')
148 for (++s; s < s_end; s++)
149 if (*s != ' ' && *s != '\t')
155 static struct pb_kexec_data *ked_prepare_data(struct ked *ked)
157 struct pb_kexec_data *kd;
160 kd = talloc(ked, struct pb_kexec_data);
165 s = ked_chomp(field_buffer(ked->fields[0], 0));
166 kd->image = *s ? talloc_strdup(kd, s) : NULL;
168 s = ked_chomp(field_buffer(ked->fields[1], 0));
169 kd->initrd = *s ? talloc_strdup(kd, s) : NULL;
171 s = ked_chomp(field_buffer(ked->fields[2], 0));
172 kd->args = *s ? talloc_strdup(kd, s) : NULL;
178 * ked_process_key - Process a user keystroke.
180 * Called from the cui via the scr:process_key method.
183 static void ked_process_key(struct nc_scr *scr)
185 struct ked *ked = ked_from_scr(scr);
192 ked_move_cursor(ked, c);
198 case 2: { /* CTRL-B */
199 struct pb_kexec_data *kd;
201 form_driver(ked->ncf, REQ_VALIDATION);
202 kd = ked_prepare_data(ked);
203 ked->on_exit(ked, ked_boot, kd);
208 ked->on_exit(ked, ked_cancel, NULL);
213 struct pb_kexec_data *kd;
215 form_driver(ked->ncf, REQ_VALIDATION);
216 kd = ked_prepare_data(ked);
217 ked->on_exit(ked, ked_update, kd);
224 ked_insert_mode_tog(ked);
229 ked_move_field(ked, REQ_FIRST_FIELD);
232 ked_move_field(ked, REQ_LAST_FIELD);
235 ked_move_field(ked, REQ_NEXT_FIELD);
238 ked_move_field(ked, REQ_PREV_FIELD);
243 ked_move_cursor(ked, REQ_BEG_FIELD);
246 ked_move_cursor(ked, REQ_END_FIELD);
249 ked_move_cursor(ked, REQ_LEFT_CHAR);
252 ked_move_cursor(ked, REQ_RIGHT_CHAR);
255 ked_move_cursor(ked, REQ_LEFT_CHAR);
256 ked_move_cursor(ked, REQ_DEL_CHAR);
259 ked_move_cursor(ked, REQ_DEL_CHAR);
266 * ked_destructor - The talloc destructor for a ked.
269 static int ked_destructor(void *arg)
271 struct ked *ked = ked_from_arg(arg);
274 for (f = ked->fields; *f; f++)
278 ked->scr.sig = pb_removed_sig;
283 static FIELD *ked_setup_field(unsigned int y, unsigned int x, char *str)
287 f = new_field(1, COLS - 1 - x, y, x, 0, 0);
288 field_opts_off(f, O_STATIC | O_WRAP);
289 set_max_field(f, 256);
290 set_field_buffer(f, 0, str);
291 set_field_status(f, 0);
295 static FIELD *ked_setup_label(unsigned int y, unsigned int x, char *str)
299 f = new_field(1, strlen(str), y, x, 0, 0);
300 field_opts_off(f, O_ACTIVE);
301 set_field_buffer(f, 0, str);
305 struct ked *ked_init(void *ui_ctx, const struct pb_kexec_data *kd,
306 void (*on_exit)(struct ked *, enum ked_result, struct pb_kexec_data *))
310 pb_log("%s: image: '%s'\n", __func__, kd->image);
311 pb_log("%s: initrd: '%s'\n", __func__, kd->initrd);
312 pb_log("%s: args: '%s'\n", __func__, kd->args);
316 ked = talloc_zero(ui_ctx, struct ked);
321 talloc_set_destructor(ked, ked_destructor);
323 nc_scr_init(&ked->scr, pb_ked_sig, 0, ui_ctx, ked_process_key,
324 ked_post, ked_unpost, ked_resize);
326 ked->scr.frame.title = talloc_strdup(ked, "Petitboot Option Editor");
327 ked->scr.frame.help = talloc_strdup(ked,
328 "ESC=cancel, Enter=accept, Ctrl-b=boot");
330 ked->on_exit = on_exit;
332 ked->fields = talloc_array(ked, FIELD *, 7);
334 ked->fields[0] = ked_setup_field(0, 9, kd->image);
335 ked->fields[1] = ked_setup_field(1, 9, kd->initrd);
336 ked->fields[2] = ked_setup_field(2, 9, kd->args);
337 ked->fields[3] = ked_setup_label(0, 1, "image:");
338 ked->fields[4] = ked_setup_label(1, 1, "initrd:");
339 ked->fields[5] = ked_setup_label(2, 1, "args:");
340 ked->fields[6] = NULL;
342 ked->ncf = new_form(ked->fields);
344 set_form_win(ked->ncf, ked->scr.main_ncw);
345 set_form_sub(ked->ncf, ked->scr.sub_ncw);