ui/ncurses: Add nc-subset selection screen

[petitboot] / discover / pxe-parser.c

diff --git a/discover/pxe-parser.c b/discover/pxe-parser.c
index 17ae55e6a3dc0c5665be7e7678a308dddf6ecf0d..95547c389799ab06ed7dc4f5a06f937e9ed4743e 100644 (file)
--- a/discover/pxe-parser.c
+++ b/discover/pxe-parser.c
@@ -2,17 +2,21 @@
 #if defined(HAVE_CONFIG_H)
 #include "config.h"
 #endif
+#include <stdlib.h>
 #include <string.h>
 
 #include <talloc/talloc.h>
 #include <url/url.h>
+#include <log/log.h>
 
 #include "parser.h"
 #include "parser-conf.h"
 #include "parser-utils.h"
 #include "resource.h"
 #include "paths.h"
+#include "event.h"
 #include "user-event.h"
+#include "network.h"
 
 static const char *pxelinux_prefix = "pxelinux.cfg/";
 
@@ -28,6 +32,91 @@ static void pxe_finish(struct conf_context *conf)
                discover_context_add_boot_option(conf->dc, info->opt);
 }
 
+/* We need a slightly modified version of pb_url_join, to allow for the
+ * pxelinux "::filename" syntax for absolute URLs
+ */
+static struct pb_url *pxe_url_join(void *ctx, const struct pb_url *url,
+               const char *s)
+{
+       struct pb_url *new_url;
+       int len;
+
+       len = strlen(s);
+
+       if (len > 2 && s[0] == ':' && s[1] == ':') {
+               char *tmp;
+
+               if (s[2] == '/') {
+                       /* ::/path -> /path */
+                       tmp = talloc_strdup(ctx, s+2);
+               } else {
+                       /* ::path -> /path */
+                       tmp = talloc_strdup(ctx, s+1);
+                       tmp[0] = '/';
+               }
+
+               new_url = pb_url_join(ctx, url, tmp);
+
+               talloc_free(tmp);
+
+       } else {
+               const char *tmp;
+               /* strip leading slashes */
+               for (tmp = s; *tmp == '/'; tmp++)
+                       ;
+               new_url = pb_url_join(ctx, url, tmp);
+       }
+
+       return new_url;
+}
+
+static void pxe_append_string(struct discover_boot_option *opt,
+               const char *str)
+{
+       if (opt->option->boot_args)
+               opt->option->boot_args = talloc_asprintf_append(
+                               opt->option->boot_args, " %s", str);
+       else
+               opt->option->boot_args = talloc_strdup(opt->option, str);
+}
+
+static char *pxe_sysappend_mac(void *ctx, uint8_t *mac)
+{
+       if (!mac)
+               return NULL;
+
+       return talloc_asprintf(ctx,
+               "BOOTIF=01-%02x-%02x-%02x-%02x-%02x-%02x",
+               mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
+}
+
+static void pxe_process_sysappend(struct discover_context *ctx,
+               struct discover_boot_option *opt,
+               unsigned long val)
+{
+       struct event *event = ctx->event;
+       char *str = NULL;
+
+       if (!event)
+               return;
+
+       if (val & 0x2) {
+               uint8_t *mac = find_mac_by_name(ctx, ctx->network,
+                                       event->device);
+               str = pxe_sysappend_mac(ctx, mac);
+               if (str) {
+                       pxe_append_string(opt, str);
+                       talloc_free(str);
+               }
+               val &= ~0x2;
+       }
+
+       if (val)
+               pb_log("pxe: unsupported features requested in "
+                               "ipappend/sysappend: 0x%04lx", val);
+
+}
+
 static void pxe_process_pair(struct conf_context *ctx,
                const char *name, char *value)
 {
@@ -72,17 +161,17 @@ static void pxe_process_pair(struct conf_context *ctx,
                return;
 
        if (streq(name, "KERNEL")) {
-               url = pb_url_join(ctx->dc, ctx->dc->conf_url, value);
+               url = pxe_url_join(ctx->dc, ctx->dc->conf_url, value);
                opt->boot_image = create_url_resource(opt, url);
 
        } else if (streq(name, "INITRD")) {
-               url = pb_url_join(ctx->dc, ctx->dc->conf_url, value);
+               url = pxe_url_join(ctx->dc, ctx->dc->conf_url, value);
                opt->initrd = create_url_resource(opt, url);
 
        } else if (streq(name, "APPEND")) {
                char *str, *end;
 
-               opt->option->boot_args = talloc_strdup(opt->option, value);
+               pxe_append_string(opt, value);
 
                str = strcasestr(value, "INITRD=");
                if (str) {
@@ -90,9 +179,16 @@ static void pxe_process_pair(struct conf_context *ctx,
                        end = strchrnul(str, ' ');
                        *end = '\0';
 
-                       url = pb_url_join(ctx->dc, ctx->dc->conf_url, str);
+                       url = pxe_url_join(ctx->dc, ctx->dc->conf_url, str);
                        opt->initrd = create_url_resource(opt, url);
                }
+       } else if (streq(name, "SYSAPPEND") || streq(name, "IPAPPEND")) {
+               unsigned long type;
+               char *end;
+
+               type = strtoul(value, &end, 10);
+               if (end != value && !(*end))
+                       pxe_process_sysappend(ctx->dc, opt, type);
        }
 
 }