]> git.ozlabs.org Git - petitboot/blobdiff - ui/ncurses/nc-config.c
ui/ncurses/nc-config: Draw config form into a scrollable pad
[petitboot] / ui / ncurses / nc-config.c
index 0b2a8d7b06738f60185b70c1b94615b1a460ad81..0f357024a0f3376af47dd9c2e5ec8a400442b88b 100644 (file)
 
 #define _GNU_SOURCE
 
-
+#include <errno.h>
+#include <stdlib.h>
 #include <string.h>
 
+#include <pb-config/pb-config.h>
 #include <talloc/talloc.h>
 #include <types/types.h>
 #include <log/log.h>
@@ -29,7 +31,7 @@
 #include "nc-config.h"
 #include "nc-widgets.h"
 
-#define N_FIELDS       18
+#define N_FIELDS       19
 
 enum net_conf_type {
        NET_CONF_TYPE_DHCP_ALL,
@@ -41,13 +43,19 @@ struct config_screen {
        struct nc_scr           scr;
        struct cui              *cui;
        struct nc_widgetset     *widgetset;
+       WINDOW                  *pad;
+
        bool                    exit;
        void                    (*on_exit)(struct cui *);
 
+       int                     scroll_y;
+
        int                     label_x;
        int                     field_x;
        int                     network_config_y;
 
+       enum net_conf_type      net_conf_type;
+
        struct {
                struct nc_widget_checkbox       *autoboot_f;
                struct nc_widget_label          *autoboot_l;
@@ -67,6 +75,7 @@ struct config_screen {
                struct nc_widget_textbox        *gateway_f;
                struct nc_widget_label          *dns_l;
                struct nc_widget_textbox        *dns_f;
+               struct nc_widget_label          *dns_help_l;
 
                struct nc_widget_button         *ok_b;
                struct nc_widget_button         *cancel_b;
@@ -84,6 +93,16 @@ static struct config_screen *config_screen_from_scr(struct nc_scr *scr)
        return config_screen;
 }
 
+static void pad_refresh(struct config_screen *screen)
+{
+       int y, x, rows, cols;
+
+       getmaxyx(screen->scr.sub_ncw, rows, cols);
+       getbegyx(screen->scr.sub_ncw, y, x);
+
+       prefresh(screen->pad, screen->scroll_y, 0, y, x, rows, cols);
+}
+
 static void config_screen_process_key(struct nc_scr *scr, int key)
 {
        struct config_screen *screen = config_screen_from_scr(scr);
@@ -93,7 +112,7 @@ static void config_screen_process_key(struct nc_scr *scr, int key)
        if (screen->exit)
                screen->on_exit(screen->cui);
        else if (handled)
-               wrefresh(screen->scr.main_ncw);
+               pad_refresh(screen);
 }
 
 static void config_screen_resize(struct nc_scr *scr)
@@ -107,7 +126,7 @@ static int config_screen_post(struct nc_scr *scr)
        struct config_screen *screen = config_screen_from_scr(scr);
        widgetset_post(screen->widgetset);
        nc_scr_frame_draw(scr);
-       wrefresh(scr->main_ncw);
+       pad_refresh(screen);
        return 0;
 }
 
@@ -123,11 +142,111 @@ struct nc_scr *config_screen_scr(struct config_screen *screen)
        return &screen->scr;
 }
 
+static int screen_process_form(struct config_screen *screen)
+{
+       const struct system_info *sysinfo = screen->cui->sysinfo;
+       struct config *config = talloc_zero(screen, struct config);
+       enum net_conf_type net_conf_type;
+       struct interface_config *iface;
+       char *str, *end;
+       int rc;
+
+       config_set_defaults(config);
+
+       config->autoboot_enabled =
+               widget_checkbox_get_value(screen->widgets.autoboot_f);
+
+
+       str = widget_textbox_get_value(screen->widgets.timeout_f);
+       if (str) {
+               unsigned long x;
+               errno = 0;
+               x = strtoul(str, &end, 10);
+               if (!errno && end != str)
+                       config->autoboot_timeout_sec = x;
+       }
+
+       net_conf_type = widget_select_get_value(screen->widgets.network_f);
+
+       /* if we don't have any network interfaces, prevent per-interface
+        * configuration */
+       if (sysinfo->n_interfaces == 0)
+               net_conf_type = NET_CONF_TYPE_DHCP_ALL;
+
+       if (net_conf_type == NET_CONF_TYPE_DHCP_ALL) {
+               config->network.n_interfaces = 0;
+
+       } else {
+               int idx;
+
+               iface = talloc_zero(config, struct interface_config);
+               config->network.n_interfaces = 1;
+               config->network.interfaces = talloc_array(config,
+                               struct interface_config *, 1);
+               config->network.interfaces[0] = iface;
+
+               /* copy hwaddr (from the sysinfo interface data) to
+                * the configuration */
+               idx = widget_select_get_value(screen->widgets.iface_f);
+               memcpy(iface->hwaddr, sysinfo->interfaces[idx]->hwaddr,
+                               sizeof(iface->hwaddr));
+       }
+
+       if (net_conf_type == NET_CONF_TYPE_DHCP_ONE) {
+               iface->method = CONFIG_METHOD_DHCP;
+       }
+
+       if (net_conf_type == NET_CONF_TYPE_STATIC) {
+               iface->method = CONFIG_METHOD_STATIC;
+               iface->static_config.address = talloc_asprintf(iface, "%s/%s",
+                               widget_textbox_get_value(
+                                       screen->widgets.ip_addr_f),
+                               widget_textbox_get_value(
+                                       screen->widgets.ip_mask_f));
+               iface->static_config.gateway = talloc_strdup(iface,
+                               widget_textbox_get_value(
+                                       screen->widgets.gateway_f));
+       }
+
+       str = widget_textbox_get_value(screen->widgets.dns_f);
+       if (str && strlen(str)) {
+               char *dns, *tmp;
+               int i;
+
+               for (;;) {
+                       dns = strtok_r(str, " \t", &tmp);
+
+                       if (!dns)
+                               break;
+
+                       i = config->network.n_dns_servers++;
+                       config->network.dns_servers = talloc_realloc(config,
+                                       config->network.dns_servers,
+                                       const char *,
+                                       config->network.n_dns_servers);
+                       config->network.dns_servers[i] =
+                               talloc_strdup(config, dns);
+
+                       str = NULL;
+               }
+       }
+
+       rc = cui_send_config(screen->cui, config);
+       talloc_free(config);
+
+       if (rc)
+               pb_log("cui_send_config failed!\n");
+       else
+               pb_debug("config sent!\n");
+
+       return 0;
+}
+
 static void ok_click(void *arg)
 {
        struct config_screen *screen = arg;
-       /* todo: save config */
-       screen->on_exit(screen->cui);
+       screen_process_form(screen);
+       screen->exit = true;
 }
 
 static void cancel_click(void *arg)
@@ -210,15 +329,21 @@ static void config_screen_layout_widgets(struct config_screen *screen,
        widget_set_visible(wf, show);
 
        if (show)
-               y += layout_pair(screen, y, screen->widgets.gateway_l, wf);
+               y += layout_pair(screen, y, screen->widgets.gateway_l, wf) + 1;
+
+       y += layout_pair(screen, y, screen->widgets.dns_l,
+                       widget_textbox_base(screen->widgets.dns_f));
 
-       wl = widget_label_base(screen->widgets.dns_l);
-       wf = widget_textbox_base(screen->widgets.dns_f);
+       /* we show the DNS/DHCP help if we're configuring DHCP */
+       show = net_conf != NET_CONF_TYPE_STATIC;
+       wl = widget_label_base(screen->widgets.dns_help_l);
        widget_set_visible(wl, show);
-       widget_set_visible(wf, show);
+       if (show) {
+               widget_move(wl, y, screen->field_x);
+               y += 1;
+       }
 
-       if (show)
-               y += 1 + layout_pair(screen, y, screen->widgets.dns_l, wf);
+       y += 1;
 
        widget_move(widget_button_base(screen->widgets.ok_b),
                        y, screen->field_x);
@@ -229,22 +354,69 @@ static void config_screen_layout_widgets(struct config_screen *screen,
 static void config_screen_network_change(void *arg, int value)
 {
        struct config_screen *screen = arg;
+       screen->net_conf_type = value;
        widgetset_unpost(screen->widgetset);
        config_screen_layout_widgets(screen, value);
        widgetset_post(screen->widgetset);
 }
 
+static struct interface_config *first_active_interface(
+               const struct config *config)
+{
+       unsigned int i;
+
+       for (i = 0; i < config->network.n_interfaces; i++) {
+               if (config->network.interfaces[i]->ignore)
+                       continue;
+               return config->network.interfaces[i];
+       }
+       return NULL;
+}
+
+static enum net_conf_type find_net_conf_type(const struct config *config)
+{
+       struct interface_config *ifcfg;
+
+       ifcfg = first_active_interface(config);
+
+       if (!ifcfg)
+               return NET_CONF_TYPE_DHCP_ALL;
+
+       else if (ifcfg->method == CONFIG_METHOD_DHCP)
+               return NET_CONF_TYPE_DHCP_ONE;
+
+       else if (ifcfg->method == CONFIG_METHOD_STATIC)
+               return NET_CONF_TYPE_STATIC;
+
+       assert(0);
+       return NET_CONF_TYPE_DHCP_ALL;
+}
+
+static void config_screen_setup_empty(struct config_screen *screen)
+{
+       widget_new_label(screen->widgetset, 2, screen->field_x,
+                       "Waiting for configuration data...");
+       screen->widgets.cancel_b = widget_new_button(screen->widgetset,
+                       4, screen->field_x, 6, "Cancel", cancel_click, screen);
+}
+
+
 static void config_screen_setup_widgets(struct config_screen *screen,
                const struct config *config,
                const struct system_info *sysinfo)
 {
        struct nc_widgetset *set = screen->widgetset;
+       struct interface_config *ifcfg;
+       char *str, *ip, *mask, *gw;
+       enum net_conf_type type;
        unsigned int i;
-       char *str;
 
        build_assert(sizeof(screen->widgets) / sizeof(struct widget *)
                        == N_FIELDS);
 
+       type = screen->net_conf_type;
+       ifcfg = first_active_interface(config);
+
        screen->widgets.autoboot_l = widget_new_label(set, 0, 0, "Autoboot:");
        screen->widgets.autoboot_f = widget_new_checkbox(set, 0, 0,
                                        config->autoboot_enabled);
@@ -259,15 +431,15 @@ static void config_screen_setup_widgets(struct config_screen *screen,
        widget_select_add_option(screen->widgets.network_f,
                                        NET_CONF_TYPE_DHCP_ALL,
                                        "DHCP on all active interfaces",
-                                       true);
+                                       type == NET_CONF_TYPE_DHCP_ALL);
        widget_select_add_option(screen->widgets.network_f,
                                        NET_CONF_TYPE_DHCP_ONE,
                                        "DHCP on a specific interface",
-                                       false);
+                                       type == NET_CONF_TYPE_DHCP_ONE);
        widget_select_add_option(screen->widgets.network_f,
                                        NET_CONF_TYPE_STATIC,
                                        "Static IP configuration",
-                                       false);
+                                       type == NET_CONF_TYPE_STATIC);
 
        widget_select_on_change(screen->widgets.network_f,
                        config_screen_network_change, screen);
@@ -277,20 +449,50 @@ static void config_screen_setup_widgets(struct config_screen *screen,
 
        for (i = 0; i < sysinfo->n_interfaces; i++) {
                struct interface_info *info = sysinfo->interfaces[i];
+               bool is_default;
+
+               is_default = ifcfg && !memcmp(ifcfg->hwaddr, info->hwaddr,
+                                       sizeof(ifcfg->hwaddr));
+
                widget_select_add_option(screen->widgets.iface_f,
-                                               i, info->name, false);
+                                               i, info->name, is_default);
+       }
+
+       gw = ip = mask = NULL;
+       if (ifcfg && ifcfg->method == CONFIG_METHOD_STATIC) {
+               char *sep;
+
+               str = talloc_strdup(screen, ifcfg->static_config.address);
+               sep = strchr(str, '/');
+               ip = str;
+
+               if (sep) {
+                       *sep = '\0';
+                       mask = sep + 1;
+               }
+               gw = ifcfg->static_config.gateway;
        }
 
        screen->widgets.ip_addr_l = widget_new_label(set, 0, 0, "IP/mask:");
-       screen->widgets.ip_addr_f = widget_new_textbox(set, 0, 0, 16, "");
+       screen->widgets.ip_addr_f = widget_new_textbox(set, 0, 0, 16, ip);
        screen->widgets.ip_mask_l = widget_new_label(set, 0, 0, "/");
-       screen->widgets.ip_mask_f = widget_new_textbox(set, 0, 0, 3, "");
+       screen->widgets.ip_mask_f = widget_new_textbox(set, 0, 0, 3, mask);
 
        screen->widgets.gateway_l = widget_new_label(set, 0, 0, "Gateway:");
-       screen->widgets.gateway_f = widget_new_textbox(set, 0, 0, 16, "");
+       screen->widgets.gateway_f = widget_new_textbox(set, 0, 0, 16, gw);
+
+       str = talloc_strdup(screen, "");
+       for (i = 0; i < config->network.n_dns_servers; i++) {
+               str = talloc_asprintf_append(str, "%s%s",
+                               (i == 0) ? "" : " ",
+                               config->network.dns_servers[i]);
+       }
 
-       screen->widgets.dns_l = widget_new_label(set, 0, 0, "DNS Server:");
-       screen->widgets.dns_f = widget_new_textbox(set, 0, 0, 16, "");
+       screen->widgets.dns_l = widget_new_label(set, 0, 0, "DNS Server(s):");
+       screen->widgets.dns_f = widget_new_textbox(set, 0, 0, 32, str);
+
+       screen->widgets.dns_help_l = widget_new_label(set, 0, 0,
+                       "(if not provided by DHCP server)");
 
        screen->widgets.ok_b = widget_new_button(set, 0, 0, 6, "OK",
                        ok_click, screen);
@@ -298,6 +500,85 @@ static void config_screen_setup_widgets(struct config_screen *screen,
                        cancel_click, screen);
 }
 
+static void config_screen_widget_focus(struct nc_widget *widget, void *arg)
+{
+       struct config_screen *screen = arg;
+       int w_y, w_height, s_max;
+
+       w_y = widget_y(widget);
+       w_height = widget_height(widget);
+       s_max = getmaxy(screen->scr.sub_ncw);
+
+       if (w_y < screen->scroll_y)
+               screen->scroll_y = w_y;
+
+       else if (w_y + w_height + screen->scroll_y > s_max - 1)
+               screen->scroll_y = 1 + w_y + w_height - s_max;
+
+       else
+               return;
+
+       pad_refresh(screen);
+}
+
+
+void config_screen_update(struct config_screen *screen,
+               const struct config *config,
+               const struct system_info *sysinfo)
+{
+       bool repost = false;
+       int height;
+
+       /* The size of the pad we'll need depends on the number of interfaces.
+        *
+        * We use N_FIELDS (which is quite conservative, as some fields share
+        * a line) as a base, then add 3 (as the network select field is
+        * takes 3 lines), and n_interfaces (as the network interface field
+        * has n_interfaces lines).
+        */
+       height = N_FIELDS + 3;
+       if (sysinfo)
+               height += sysinfo->n_interfaces;
+       if (!screen->pad || getmaxx(screen->pad) < height) {
+               if (screen->pad)
+                       delwin(screen->pad);
+               screen->pad = newpad(height, COLS);
+       }
+
+       if (screen->widgetset) {
+               widgetset_unpost(screen->widgetset);
+               talloc_free(screen->widgetset);
+               repost = true;
+       }
+
+       screen->widgetset = widgetset_create(screen, screen->scr.main_ncw,
+                       screen->pad);
+       widgetset_set_widget_focus(screen->widgetset,
+                       config_screen_widget_focus, screen);
+
+       if (!config || !sysinfo) {
+               config_screen_setup_empty(screen);
+       } else {
+               screen->net_conf_type = find_net_conf_type(config);
+
+               config_screen_setup_widgets(screen, config, sysinfo);
+               config_screen_layout_widgets(screen, screen->net_conf_type);
+       }
+
+       if (repost)
+               widgetset_post(screen->widgetset);
+
+       pad_refresh(screen);
+}
+
+static int config_screen_destroy(void *arg)
+{
+       struct config_screen *screen = arg;
+       if (screen->pad)
+               delwin(screen->pad);
+       return 0;
+}
+
 struct config_screen *config_screen_init(struct cui *cui,
                const struct config *config,
                const struct system_info *sysinfo,
@@ -306,6 +587,7 @@ struct config_screen *config_screen_init(struct cui *cui,
        struct config_screen *screen;
 
        screen = talloc_zero(cui, struct config_screen);
+       talloc_set_destructor(screen, config_screen_destroy);
        nc_scr_init(&screen->scr, pb_config_screen_sig, 0,
                        cui, config_screen_process_key,
                        config_screen_post, config_screen_unpost,
@@ -314,7 +596,7 @@ struct config_screen *config_screen_init(struct cui *cui,
        screen->cui = cui;
        screen->on_exit = on_exit;
        screen->label_x = 2;
-       screen->field_x = 16;
+       screen->field_x = 17;
 
        screen->scr.frame.ltitle = talloc_strdup(screen,
                        "Petitboot System Configuration");
@@ -323,13 +605,11 @@ struct config_screen *config_screen_init(struct cui *cui,
                        "tab=next, shift+tab=previous");
        nc_scr_frame_draw(&screen->scr);
 
-       screen->widgetset = widgetset_create(screen, screen->scr.main_ncw,
-                       screen->scr.sub_ncw);
-       config_screen_setup_widgets(screen, config, sysinfo);
-       config_screen_layout_widgets(screen, NET_CONF_TYPE_DHCP_ALL);
+       scrollok(screen->scr.sub_ncw, true);
+
+       config_screen_update(screen, config, sysinfo);
 
        wrefresh(screen->scr.main_ncw);
-       scrollok(screen->scr.sub_ncw, true);
 
        return screen;
 }