Optionally include the install command as found on Juniper products

like EX and SRX. The install command uses pkgfs to extract a kernel,
zero or more modules and a root file system from the specified package
and boots the kernel. The name of the kernel, the list of modules and
the name of the root file system can be specified by putting a
file called "metatags in the package.

The package to use is given by an URL. The schemes supported are
tftp and file. For the file scheme, the disk is currently hardcoded
but that should really look for the package on all devices and
partititions.

Obtained from:	Juniper Networks, Inc.

sys/boot/common/Makefile.inc

@@ -72,4 +72,9 @@ MAN+=	../forth/version.4th.8
 CFLAGS+=	-DBOOT_PROMPT_123
 .endif
 
+.if defined(LOADER_INSTALL_SUPPORT)
+SRCS+=	install.c
+CFLAGS+=-I${.CURDIR}/../../../../lib/libstand
+.endif
+
 MAN+=	loader.8

sys/boot/common/install.c

@@ -0,0 +1,339 @@
+/*-
+ * Copyright (c) 2008-2014, Juniper Networks, Inc.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
+ * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
+ * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ */
+
+#include <sys/cdefs.h>
+__FBSDID("$FreeBSD$");
+
+#include <sys/param.h>
+#include <sys/socket.h>
+#include <net/if.h>
+#include <netinet/in.h>
+#include <netinet/in_systm.h>
+
+#include <stand.h>
+#include <net.h>
+#include <string.h>
+
+#include "bootstrap.h"
+
+extern struct in_addr rootip;
+extern struct in_addr servip;
+
+extern int pkgfs_init(const char *, struct fs_ops *);
+extern void pkgfs_cleanup(void);
+
+COMMAND_SET(install, "install", "install software package", command_install);
+
+static char *inst_kernel;
+static char **inst_modules;
+static char *inst_rootfs;
+
+static int
+setpath(char **what, char *val)
+{
+	char *path;
+	size_t len;
+	int rel;
+
+	len = strlen(val) + 1;
+	rel = (val[0] != '/') ? 1 : 0;
+	path = malloc(len + rel);
+	if (path == NULL)
+		return (ENOMEM);
+	path[0] = '/';
+	strcpy(path + rel, val);
+
+	*what = path;
+	return (0);
+}
+
+static int
+setmultipath(char ***what, char *val)
+{
+	char *s, *v;
+	int count, error, idx;
+
+	count = 0;
+	v = val;
+	do {
+		count++;
+		s = strchr(v, ',');
+		v = (s == NULL) ? NULL : s + 1;
+	} while (v != NULL);
+
+	*what = calloc(count + 1, sizeof(char *));
+	if (*what == NULL)
+		return (ENOMEM);
+
+	for (idx = 0; idx < count; idx++) {
+		s = strchr(val, ',');
+		if (s != NULL)
+			*s++ = '\0';
+		error = setpath(*what + idx, val);
+		if (error)
+			return (error);
+		val = s;
+	}
+
+	return (0);
+}
+
+static int
+read_metatags(int fd)
+{
+	char buf[1024];
+	char *p, *tag, *val;
+	ssize_t fsize;
+	int error;
+
+	fsize = read(fd, buf, sizeof(buf));
+	if (fsize == -1)
+		return (errno);
+
+	/*
+	 * Assume that if we read a whole buffer worth of data, we
+	 * haven't read the entire file. In other words, the buffer
+	 * size must always be larger than the file size. That way
+	 * we can append a '\0' and use standard string operations.
+	 * Return an error if this is not possible.
+	 */
+	if (fsize == sizeof(buf))
+		return (ENOMEM);
+
+	buf[fsize] = '\0';
+	error = 0;
+	tag = buf;
+	while (!error && *tag != '\0') {
+		val = strchr(tag, '=');
+		if (val == NULL) {
+			error = EINVAL;
+			break;
+		}
+		*val++ = '\0';
+		p = strchr(val, '\n');
+		if (p == NULL) {
+			error = EINVAL;
+			break;
+		}
+		*p++ = '\0';
+
+		if (strcmp(tag, "KERNEL") == 0)
+			error = setpath(&inst_kernel, val);
+		else if (strcmp(tag, "MODULES") == 0)
+			error = setmultipath(&inst_modules, val);
+		else if (strcmp(tag, "ROOTFS") == 0)
+			error = setpath(&inst_rootfs, val);
+
+		tag = p;
+	}
+
+	return (error);
+}
+
+static void
+cleanup(void)
+{
+	u_int i;
+
+	if (inst_kernel != NULL) {
+		free(inst_kernel);
+		inst_kernel = NULL;
+	}
+	if (inst_modules != NULL) {
+		i = 0;
+		while (inst_modules[i] != NULL)
+			free(inst_modules[i++]);
+		free(inst_modules);
+		inst_modules = NULL;
+	}
+	if (inst_rootfs != NULL) {
+		free(inst_rootfs);
+		inst_rootfs = NULL;
+	}
+	pkgfs_cleanup();
+}
+
+/*
+ * usage: install URL
+ * where: URL = (tftp|file)://[host]/<package>
+ */
+static int
+install(char *pkgname)
+{
+	static char buf[256];
+	struct fs_ops *proto;
+	struct preloaded_file *fp;
+	char *s, *currdev;
+	const char *devname;
+	int error, fd, i, local;
+
+	s = strstr(pkgname, "://");
+	if (s == NULL)
+		goto invalid_url;
+
+	i = s - pkgname;
+	if (i == 4 && !strncasecmp(pkgname, "tftp", i)) {
+		devname = "net0";
+		proto = &tftp_fsops;
+		local = 0;
+	} else if (i == 4 && !strncasecmp(pkgname, "file", i)) {
+		currdev = getenv("currdev");
+		if (currdev != NULL && strcmp(currdev, "pxe0:") == 0) {
+			devname = "pxe0";
+			proto = NULL;
+		} else {
+			devname = "disk1";
+			proto = &dosfs_fsops;
+		}
+		local = 1;
+	} else
+		goto invalid_url;
+
+	s += 3;
+	if (*s == '\0')
+		goto invalid_url;
+
+	if (*s != '/' ) {
+		if (local)
+			goto invalid_url;
+
+		pkgname = strchr(s, '/');
+		if (pkgname == NULL)
+			goto invalid_url;
+
+		*pkgname = '\0';
+		servip.s_addr = inet_addr(s);
+		if (servip.s_addr == htonl(INADDR_NONE))
+			goto invalid_url;
+
+		setenv("serverip", inet_ntoa(servip), 1);
+
+		*pkgname = '/';
+	} else
+		pkgname = s;
+
+	if (strlen(devname) + strlen(pkgname) + 2 > sizeof(buf)) {
+		command_errmsg = "package name too long";
+		return (CMD_ERROR);
+	}
+	sprintf(buf, "%s:%s", devname, pkgname);
+	setenv("install_package", buf, 1);
+
+	error = pkgfs_init(buf, proto);
+	if (error) {
+		command_errmsg = "cannot open package";
+		goto fail;
+	}
+
+	/*
+	 * Point of no return: unload anything that may have been
+	 * loaded and prune the environment from harmful variables.
+	 */
+	unload();
+	unsetenv("vfs.root.mountfrom");
+
+	/*
+	 * read the metatags file.
+	 */
+	fd = open("/metatags", O_RDONLY);
+	if (fd != -1) {
+		error = read_metatags(fd);
+		close(fd);
+		if (error) {
+			command_errmsg = "cannot load metatags";
+			goto fail;
+		}
+	}
+
+	s = (inst_kernel == NULL) ? "/kernel" : inst_kernel;
+	error = mod_loadkld(s, 0, NULL);
+	if (error) {
+		command_errmsg = "cannot load kernel from package";
+		goto fail;
+	}
+
+	i = 0;
+	while (inst_modules != NULL && inst_modules[i] != NULL) {
+		error = mod_loadkld(inst_modules[i], 0, NULL);
+		if (error) {
+			command_errmsg = "cannot load module(s) from package";
+			goto fail;
+		}
+		i++;
+	}
+
+	s = (inst_rootfs == NULL) ? "/install.iso" : inst_rootfs;
+	if (file_loadraw("mfs_root", s) == NULL) {
+		error = errno;
+		command_errmsg = "cannot load root file system";
+		goto fail;
+	}
+
+	cleanup();
+
+	fp = file_findfile(NULL, NULL);
+	if (fp != NULL)
+		file_formats[fp->f_loader]->l_exec(fp);
+	error = CMD_ERROR;
+	command_errmsg = "unable to start installation";
+
+ fail:
+	sprintf(buf, "%s (error %d)", command_errmsg, error);
+	cleanup();
+	unload();
+	exclusive_file_system = NULL;
+	command_errmsg = buf;	/* buf is static. */
+	return (CMD_ERROR);
+
+ invalid_url:
+	command_errmsg = "invalid URL";
+	return (CMD_ERROR);
+}
+
+static int
+command_install(int argc, char *argv[])
+{
+	int argidx;
+
+	unsetenv("install_format");
+
+	argidx = 1;
+	while (1) {
+		if (argc == argidx) {
+			command_errmsg =
+			    "usage: install [--format] <URL>";
+			return (CMD_ERROR);
+		}
+		if (!strcmp(argv[argidx], "--format")) {
+			setenv("install_format", "yes", 1);
+			argidx++;
+			continue;
+		}
+		break;
+	}
+
+	return (install(argv[argidx]));
+}