X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=ui%2Fncurses%2Fnc-boot-editor.c;h=8fe33656441d07a24c5be75dc8332a96b13c52b7;hp=fb0bee8ef526a9d4c6ab853072388ff13b793cef;hb=6d1eff1e8e1db12e7aa179e8dc9c65050c045536;hpb=a610837ff38f5cc80bcbad465a80ab920e67927d;ds=inline

diff --git a/ui/ncurses/nc-boot-editor.c b/ui/ncurses/nc-boot-editor.c
index fb0bee8..8fe3365 100644
--- a/ui/ncurses/nc-boot-editor.c
+++ b/ui/ncurses/nc-boot-editor.c
@@ -16,6 +16,8 @@
  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  */
 
+#include "config.h"
+
 #define _GNU_SOURCE
 
 #include <assert.h>
@@ -185,6 +187,9 @@ static struct pb_boot_data *boot_editor_prepare_data(
 	bd->initrd = *s ? talloc_strdup(bd, s) : NULL;
 
 	s = boot_editor_chomp(field_buffer(boot_editor->fields[2], 0));
+	bd->dtb = *s ? talloc_strdup(bd, s) : NULL;
+
+	s = boot_editor_chomp(field_buffer(boot_editor->fields[3], 0));
 	bd->args = *s ? talloc_strdup(bd, s) : NULL;
 
 	return bd;
@@ -208,16 +213,14 @@ static void boot_editor_process_key(struct nc_scr *scr, int key)
 
 	/* hot keys */
 	case 27: /* ESC */
-		boot_editor->on_exit(boot_editor,
-				boot_editor_cancel, NULL);
+		boot_editor->on_exit(boot_editor, boot_editor_cancel, NULL);
 		nc_flush_keys();
 		return;
 	case '\n':
 	case '\r':
 		form_driver(boot_editor->ncf, REQ_VALIDATION);
 		bd = boot_editor_prepare_data(boot_editor);
-		boot_editor->on_exit(boot_editor,
-				boot_editor_update, bd);
+		boot_editor->on_exit(boot_editor, boot_editor_update, bd);
 		nc_flush_keys();
 		return;
 
@@ -305,47 +308,57 @@ static FIELD *boot_editor_setup_label(unsigned int y, unsigned int x, char *str)
 	return f;
 }
 
-struct boot_editor *boot_editor_init(void *ui_ctx,
+struct boot_editor *boot_editor_init(struct pmenu *menu,
 		const struct pb_boot_data *bd,
 		void (*on_exit)(struct boot_editor *,
 				enum boot_editor_result,
 				struct pb_boot_data *))
 {
+	char *image, *initrd, *dtb, *args;
 	struct boot_editor *boot_editor;
 
-	pb_log("%s: image:  '%s'\n", __func__, bd->image);
-	pb_log("%s: initrd: '%s'\n", __func__, bd->initrd);
-	pb_log("%s: args:   '%s'\n", __func__, bd->args);
-
 	assert(on_exit);
 
-	boot_editor = talloc_zero(ui_ctx, struct boot_editor);
+	boot_editor = talloc_zero(menu, struct boot_editor);
 
 	if (!boot_editor)
 		return NULL;
 
 	talloc_set_destructor(boot_editor, boot_editor_destructor);
+	boot_editor->original_pmenu = menu;
 
 	nc_scr_init(&boot_editor->scr, pb_boot_editor_sig, 0,
-			ui_ctx, boot_editor_process_key,
+			menu, boot_editor_process_key,
 		boot_editor_post, boot_editor_unpost, boot_editor_resize);
 
-	boot_editor->scr.frame.title = talloc_strdup(boot_editor,
+	boot_editor->scr.frame.ltitle = talloc_strdup(boot_editor,
 			"Petitboot Option Editor");
+	boot_editor->scr.frame.rtitle = NULL;
 	boot_editor->scr.frame.help = talloc_strdup(boot_editor,
 			"ESC=cancel, Enter=accept");
 
 	boot_editor->on_exit = on_exit;
 
-	boot_editor->fields = talloc_array(boot_editor, FIELD *, 7);
+	boot_editor->fields = talloc_array(boot_editor, FIELD *, 9);
+
+	if (bd) {
+		image = bd->image;
+		initrd = bd->initrd;
+		dtb = bd->dtb;
+		args = bd->args;
+	} else {
+		image = initrd = dtb = args = "";
+	}
 
-	boot_editor->fields[0] = boot_editor_setup_field(0, 9, bd->image);
-	boot_editor->fields[1] = boot_editor_setup_field(1, 9, bd->initrd);
-	boot_editor->fields[2] = boot_editor_setup_field(2, 9, bd->args);
-	boot_editor->fields[3] = boot_editor_setup_label(0, 1, "image:");
-	boot_editor->fields[4] = boot_editor_setup_label(1, 1, "initrd:");
-	boot_editor->fields[5] = boot_editor_setup_label(2, 1, "args:");
-	boot_editor->fields[6] = NULL;
+	boot_editor->fields[0] = boot_editor_setup_field(0, 9, image);
+	boot_editor->fields[1] = boot_editor_setup_field(1, 9, initrd);
+	boot_editor->fields[2] = boot_editor_setup_field(2, 9, dtb);
+	boot_editor->fields[3] = boot_editor_setup_field(3, 9, args);
+	boot_editor->fields[4] = boot_editor_setup_label(0, 1, "image:");
+	boot_editor->fields[5] = boot_editor_setup_label(1, 1, "initrd:");
+	boot_editor->fields[6] = boot_editor_setup_label(2, 1, "dtb:");
+	boot_editor->fields[7] = boot_editor_setup_label(3, 1, "args:");
+	boot_editor->fields[8] = NULL;
 
 	boot_editor->ncf = new_form(boot_editor->fields);