X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fuser-event.c;h=cf7f20c599bbefdac3c004834dc75966500a7a4b;hp=d7e0d195035efa43202137de78d57bfb78fd3da8;hb=922756a3aefd8ba5c5675efd37df9f43938cf432;hpb=08e967ee301b502b2114e3855210a0b2600d095c

diff --git a/discover/user-event.c b/discover/user-event.c
index d7e0d19..cf7f20c 100644
--- a/discover/user-event.c
+++ b/discover/user-event.c
@@ -20,7 +20,6 @@
 #include "config.h"
 #endif
 
-#define _GNU_SOURCE
 #include <assert.h>
 #include <errno.h>
 #include <string.h>
@@ -90,9 +89,9 @@ static struct resource *user_event_resource(struct discover_boot_option *opt,
 		return NULL;
 	}
 
-	boot_file = event_get_param(event, "boot_file");
+	boot_file = event_get_param(event, "bootfile");
 	if (!boot_file) {
-		pb_log("%s: boot_file not found\n", __func__);
+		pb_log("%s: bootfile not found\n", __func__);
 		return NULL;
 	}
 
@@ -209,7 +208,7 @@ static char *parse_mac_addr(struct discover_context *ctx, const char *mac)
 			mac_addr_arr + 2, mac_addr_arr + 3, mac_addr_arr + 4,
 			mac_addr_arr + 5);
 
-	mac_addr = talloc_asprintf(ctx, "01-%02X-%02X-%02X-%02X-%02X-%02X",
+	mac_addr = talloc_asprintf(ctx, "01-%02x-%02x-%02x-%02x-%02x-%02x",
 			mac_addr_arr[0], mac_addr_arr[1], mac_addr_arr[2],
 			mac_addr_arr[3], mac_addr_arr[4], mac_addr_arr[5]);
 
@@ -231,62 +230,85 @@ static char *parse_ip_addr(struct discover_context *ctx, const char *ip)
 }
 
 struct pb_url *user_event_parse_conf_url(struct discover_context *ctx,
-		struct event *event)
+		struct event *event, bool *is_complete)
 {
-	const char *conffile, *host, *bootfile;
+	const char *conffile, *pathprefix, *host, *bootfile;
 	char *p, *basedir, *url_str;
 	struct pb_url *url;
 
 	conffile = event_get_param(event, "pxeconffile");
-	if (conffile) {
-		if (is_url(conffile)) {
-			url = pb_url_parse(ctx, conffile);
-		} else {
-			host = parse_host_addr(event);
-			if (!host) {
-				pb_log("%s: host address not found\n",
-						__func__);
-				return NULL;
-			}
-
-			url_str = talloc_asprintf(ctx, "%s%s/%s", "tftp://",
-					host, conffile);
-			url = pb_url_parse(ctx, url_str);
+	pathprefix = event_get_param(event, "pxepathprefix");
+	bootfile = event_get_param(event, "bootfile");
+
+	/* If we're given a conf file, we're able to generate a complete URL to
+	 * the configuration file, and the parser doesn't need to do any
+	 * further autodiscovery */
+	*is_complete = !!conffile;
+
+	/* if conffile is a URL, that's all we need */
+	if (conffile && is_url(conffile)) {
+		url = pb_url_parse(ctx, conffile);
+		return url;
+	}
 
+	/* If we can create a URL from pathprefix (optionally with
+	 * conffile appended to create a complete URL), use that */
+	if (pathprefix && is_url(pathprefix)) {
+		if (conffile) {
+			url_str = talloc_asprintf(ctx, "%s%s",
+					pathprefix, conffile);
+			url = pb_url_parse(ctx, url_str);
 			talloc_free(url_str);
+		} else {
+			url = pb_url_parse(ctx, pathprefix);
 		}
 
-		ctx->conf_url = url;
-	} else {
-		host = parse_host_addr(event);
-		if (!host) {
-			pb_log("%s: host address not found\n", __func__);
-			return NULL;
-		}
+		return url;
+	}
 
-		bootfile = event_get_param(event, "bootfile");
-		if (!bootfile) {
-			pb_log("%s: bootfile param not found\n", __func__);
-			return NULL;
-		}
+	host = parse_host_addr(event);
+	if (!host) {
+		pb_log("%s: host address not found\n", __func__);
+		return NULL;
+	}
+
+	url_str = talloc_asprintf(ctx, "tftp://%s/", host);
+
+	/* if we have a pathprefix, use that directly.. */
+	if (pathprefix) {
+		/* strip leading slashes */
+		while (pathprefix[0] == '/')
+			pathprefix++;
+		url_str = talloc_asprintf_append(url_str, "%s", pathprefix);
+
+	/* ... otherwise, add a path based on the bootfile name, but only
+	 * if conffile isn't an absolute path itself */
+	} else if (bootfile && !(conffile && conffile[0] == '/')) {
 
 		basedir = talloc_strdup(ctx, bootfile);
-		p = strchr(basedir, '/');
-		if (p)
-			*p = '\0';
 
-		if (!strcmp(basedir,"") || !strcmp(basedir, "."))
-			url_str = talloc_asprintf(ctx, "%s%s/", "tftp://",host);
-		else
-			url_str = talloc_asprintf(ctx, "%s%s/%s/", "tftp://",host,
-					basedir);
+		/* strip filename from the bootfile path, leaving only a
+		 * directory */
+		p = strrchr(basedir, '/');
+		if (!p)
+			p = basedir;
+		*p = '\0';
 
-		url = pb_url_parse(ctx, url_str);
+		if (strlen(basedir))
+			url_str = talloc_asprintf_append(url_str, "%s/",
+					basedir);
 
-		talloc_free(url_str);
 		talloc_free(basedir);
 	}
 
+	/* finally, append conffile */
+	if (conffile)
+		url_str = talloc_asprintf_append(url_str, "%s", conffile);
+
+	url = pb_url_parse(ctx, url_str);
+
+	talloc_free(url_str);
+
 	return url;
 }