Rework to simplify and impose sane url syntax. That is we allow for file://[devname[:fstype]]/package Reviewed by: stevek MFC after: 1 week Sponsored by: Juniper Networks Differential Revision: https://reviews.freebsd.org//D25134
410 lines
8.5 KiB
C
410 lines
8.5 KiB
C
/*-
|
|
* 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 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 char *inst_loader_rc;
|
|
|
|
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);
|
|
else if (strcmp(tag, "LOADER_RC") == 0)
|
|
error = setpath(&inst_loader_rc, 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;
|
|
}
|
|
if (inst_loader_rc != NULL) {
|
|
free(inst_loader_rc);
|
|
inst_loader_rc = NULL;
|
|
}
|
|
pkgfs_cleanup();
|
|
}
|
|
|
|
/*
|
|
* usage: install URL
|
|
* where: URL = tftp://[host]/<package>
|
|
* or file://[devname[:fstype]]/<package>
|
|
*/
|
|
static int
|
|
install(char *pkgname)
|
|
{
|
|
static char buf[256];
|
|
struct fs_ops *proto;
|
|
struct preloaded_file *fp;
|
|
char *e, *s, *currdev;
|
|
char *devname;
|
|
size_t devnamelen;
|
|
int error, fd, i, local;
|
|
|
|
s = strstr(pkgname, "://");
|
|
if (s == NULL)
|
|
goto invalid_url;
|
|
|
|
i = s - pkgname;
|
|
s += 3;
|
|
if (*s == '\0')
|
|
goto invalid_url;
|
|
|
|
proto = NULL;
|
|
devname = NULL;
|
|
devnamelen = 0;
|
|
|
|
if (i == 4 && !strncasecmp(pkgname, "tftp", i)) {
|
|
devname = "net0";
|
|
devnamelen = 4;
|
|
proto = &tftp_fsops;
|
|
local = 0;
|
|
} else if (i == 4 && !strncasecmp(pkgname, "file", i)) {
|
|
currdev = getenv("currdev");
|
|
local = 1;
|
|
|
|
if (*s == '/') { /* file:/// */
|
|
if (devname == NULL)
|
|
devname = currdev;
|
|
if (devname == NULL)
|
|
devname = "disk1";
|
|
} else { /* file://devname[:fstype]/ */
|
|
devname = s;
|
|
e = strchr(devname, '/');
|
|
if (!e)
|
|
goto invalid_url;
|
|
devnamelen = e - devname;
|
|
s = e; /* consume devname */
|
|
}
|
|
if ((e = strchr(devname, ':')) != NULL) {
|
|
/* could be :fstype */
|
|
devnamelen = e - devname;
|
|
switch (e[1]) {
|
|
case '\0': /* just currdev */
|
|
break;
|
|
case 'd':
|
|
proto = &dosfs_fsops;
|
|
break;
|
|
#ifdef HOSTPROG
|
|
case 'h':
|
|
{
|
|
extern struct fs_ops host_fsops;
|
|
|
|
proto = &host_fsops;
|
|
}
|
|
break;
|
|
#endif
|
|
case 'u':
|
|
proto = &ufs_fsops;
|
|
break;
|
|
}
|
|
}
|
|
if (proto == NULL && strncmp(devname, "disk", 4) == 0) {
|
|
proto = &dosfs_fsops;
|
|
}
|
|
}
|
|
|
|
if (devname == NULL)
|
|
goto invalid_url;
|
|
|
|
if (devnamelen == 0) {
|
|
/* default is currdev which ends with ':' */
|
|
devnamelen = strlen(devname);
|
|
if (devname[devnamelen - 1] == ':')
|
|
devnamelen--;
|
|
}
|
|
|
|
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);
|
|
|
|
if (proto == &tftp_fsops) {
|
|
tftpip.s_addr = servip.s_addr;
|
|
}
|
|
|
|
*pkgname = '/';
|
|
} else
|
|
pkgname = s;
|
|
|
|
i = snprintf(buf, sizeof(buf), "%.*s:%s",
|
|
(int) devnamelen, devname, pkgname);
|
|
if (i >= (int) sizeof(buf)) {
|
|
command_errmsg = "package name too long";
|
|
return (CMD_ERROR);
|
|
}
|
|
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;
|
|
}
|
|
|
|
/* If there is a loader.rc in the package, execute it */
|
|
s = (inst_loader_rc == NULL) ? "/loader.rc" : inst_loader_rc;
|
|
fd = open(s, O_RDONLY);
|
|
if (fd != -1) {
|
|
close(fd);
|
|
error = inter_include(s);
|
|
if (error == CMD_ERROR)
|
|
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(s, "mfs_root", 1) == 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]));
|
|
}
|