--- /dev/null
+/*
+ * Copyright (C) 2018 IBM Corporation
+ *
+ * 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(HAVE_CONFIG_H)
+#include "config.h"
+#endif
+
+#include <errno.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include <talloc/talloc.h>
+#include <types/types.h>
+#include <i18n/i18n.h>
+#include <log/log.h>
+
+#include "nc-cui.h"
+#include "nc-widgets.h"
+#include "nc-auth.h"
+
+#define N_FIELDS 5
+
+struct auth_screen {
+ struct nc_scr scr;
+ struct cui *cui;
+ struct nc_scr *return_scr;
+ struct nc_widgetset *widgetset;
+ void (*process_key)(struct nc_scr *, int);
+
+ bool set_password;
+ void (*callback)(struct nc_scr *);
+ int offset_y;
+ int label_x;
+ int field_x;
+
+ bool exit;
+ void (*on_exit)(struct cui *);
+
+ struct {
+ struct nc_widget_label *title_a_l;
+ struct nc_widget_label *title_b_l;
+ struct nc_widget_textbox *password_f;
+ struct nc_widget_label *new_l;
+ struct nc_widget_textbox *new_f;
+ struct nc_widget_button *ok_b;
+ struct nc_widget_button *cancel_b;
+ } widgets;
+};
+
+struct nc_scr *auth_screen_return_scr(struct auth_screen *screen)
+{
+ return screen->return_scr;
+}
+
+struct nc_scr *auth_screen_scr(struct auth_screen *screen)
+{
+ return &screen->scr;
+}
+
+static struct auth_screen *auth_screen_from_scr(struct nc_scr *scr)
+{
+ struct auth_screen *auth_screen;
+
+ assert(scr->sig == pb_auth_screen_sig);
+ auth_screen = (struct auth_screen *)
+ ((char *)scr - (size_t)&((struct auth_screen *)0)->scr);
+ assert(auth_screen->scr.sig == pb_auth_screen_sig);
+ return auth_screen;
+}
+
+static void auth_screen_process_key(struct nc_scr *scr, int key)
+{
+ struct auth_screen *screen = auth_screen_from_scr(scr);
+ bool handled;
+
+ handled = widgetset_process_key(screen->widgetset, key);
+
+ if (!handled) {
+ switch (key) {
+ case 'x':
+ case 27: /* esc */
+ screen->exit = true;
+ break;
+ }
+ }
+
+ if (screen->exit)
+ screen->on_exit(screen->cui);
+ else if (handled)
+ wrefresh(screen->scr.sub_ncw);
+}
+
+static void auth_screen_frame_draw(struct nc_scr *scr)
+{
+ int y, x;
+
+ getmaxyx(scr->sub_ncw, y, x);
+
+ mvwhline(scr->sub_ncw, 0, 0, ACS_HLINE, x);
+ mvwhline(scr->sub_ncw, y - 1, 0, ACS_HLINE, x);
+
+ mvwvline(scr->sub_ncw, 0, 0, ACS_VLINE, y);
+ mvwvline(scr->sub_ncw, 0, x - 1, ACS_VLINE, y);
+}
+
+static int auth_screen_post(struct nc_scr *scr)
+{
+ struct auth_screen *screen = auth_screen_from_scr(scr);
+ widgetset_post(screen->widgetset);
+ auth_screen_frame_draw(scr);
+ wrefresh(scr->sub_ncw);
+ return 0;
+}
+
+static int auth_screen_unpost(struct nc_scr *scr)
+{
+ struct auth_screen *screen = auth_screen_from_scr(scr);
+ widgetset_unpost(screen->widgetset);
+ return 0;
+}
+
+static void ok_click(void *arg)
+{
+ struct auth_screen *screen = arg;
+ char *password, *new_password;
+ int rc;
+
+
+ password = widget_textbox_get_value(screen->widgets.password_f);
+ if (screen->set_password) {
+ new_password = widget_textbox_get_value(screen->widgets.new_f);
+ rc = cui_send_set_password(screen->cui, password, new_password);
+ } else
+ rc = cui_send_authenticate(screen->cui, password);
+
+ if (rc)
+ pb_log("Failed to send authenticate action\n");
+ else if (screen->callback)
+ screen->callback(screen->return_scr);
+
+ screen->exit = true;
+}
+
+static void cancel_click(void *arg)
+{
+ struct auth_screen *screen = arg;
+ screen->exit = true;
+}
+
+static void auth_screen_layout_widgets(struct auth_screen *screen)
+{
+ int y = 1;
+
+ widget_move(widget_label_base(screen->widgets.title_a_l),
+ y++, screen->label_x);
+ widget_move(widget_label_base(screen->widgets.title_b_l),
+ y++, screen->label_x);
+
+ y += 1;
+
+ widget_move(widget_textbox_base(screen->widgets.password_f),
+ y++, screen->field_x);
+
+ y += 1;
+
+ if (screen->set_password) {
+ widget_move(widget_label_base(screen->widgets.new_l),
+ y++, screen->label_x);
+ widget_move(widget_textbox_base(screen->widgets.new_f),
+ y++, screen->field_x);
+ y += 1;
+ }
+
+ widget_move(widget_button_base(screen->widgets.ok_b),
+ y, 10);
+ widget_move(widget_button_base(screen->widgets.cancel_b),
+ y, 30);
+}
+
+static void auth_screen_draw(struct auth_screen *screen)
+{
+ struct nc_widgetset *set;
+
+ set = widgetset_create(screen, screen->scr.main_ncw,
+ screen->scr.sub_ncw);
+ if (!set) {
+ pb_log("%s: failed to create widgetset\n", __func__);
+ return;
+ }
+ screen->widgetset = set;
+
+ screen->widgets.title_a_l = widget_new_label(set, 0, 0,
+ _("This action requires authorisation."));
+ screen->widgets.title_b_l = widget_new_label(set, 0, 0,
+ _("Please enter the system password."));
+
+ screen->widgets.password_f = widget_new_textbox_hidden(set, 0, 0,
+ COLS - 20 - 20, "", true);
+
+ if (screen->set_password) {
+ screen->widgets.new_l = widget_new_label(set, 0, 0,
+ _("New password:"));
+ screen->widgets.new_f = widget_new_textbox_hidden(set, 0, 0,
+ COLS - 20 - 20, "", true);
+ }
+
+ screen->widgets.ok_b = widget_new_button(set, 0, 0, 10, _("OK"),
+ ok_click, screen);
+ screen->widgets.cancel_b = widget_new_button(set, 0, 0, 10, _("Cancel"),
+ cancel_click, screen);
+
+ auth_screen_layout_widgets(screen);
+}
+
+static int auth_screen_destroy(void *arg)
+{
+ struct auth_screen *screen = arg;
+ if (screen->scr.sub_ncw)
+ delwin(screen->scr.sub_ncw);
+ return 0;
+}
+
+struct auth_screen *auth_screen_init(struct cui *cui,
+ WINDOW *parent, bool set_password,
+ void (*callback)(struct nc_scr *),
+ void (*on_exit)(struct cui *))
+{
+ struct auth_screen *screen = NULL;
+ struct nc_scr *scr;
+ int y, x;
+
+ if (!cui || !parent)
+ return NULL;
+
+ screen = talloc_zero(cui, struct auth_screen);
+ if (!screen)
+ return NULL;
+ talloc_set_destructor(screen, auth_screen_destroy);
+
+ screen->cui = cui;
+ screen->return_scr = cui->current;
+ screen->set_password = set_password;
+ screen->callback = callback;
+ screen->on_exit = on_exit;
+ screen->label_x = 5;
+ screen->field_x = 10;
+
+ /*
+ * Manually init our nc_scr: we only want to create the subwin and
+ * 'inherit' the parent window.
+ */
+ scr = &screen->scr;
+ scr->sig = pb_auth_screen_sig;
+ scr->ui_ctx = cui;
+ scr->process_key = auth_screen_process_key;
+ scr->post = auth_screen_post;
+ scr->unpost = auth_screen_unpost;
+ scr->resize = NULL;
+
+
+ getbegyx(parent, y, x);
+ /* Hold on to the real offset from the top of the screen */
+ screen->offset_y = y + 5;
+ (void)x;
+
+ scr->main_ncw = parent;
+ scr->sub_ncw = derwin(parent, set_password ? 15 : 10, COLS - 20,
+ 5, 10); /* relative to parent origin */
+ if (!scr->sub_ncw) {
+ pb_log("Could not create subwin\n");
+ goto err;
+ }
+
+ auth_screen_draw(screen);
+
+ return screen;
+err:
+ pb_log("failed to create auth screen\n");
+ if (screen) {
+ if (screen->scr.sub_ncw)
+ delwin(screen->scr.sub_ncw);
+ talloc_free(screen);
+ }
+ return NULL;
+}
projects / petitboot / blobdiff