userboot/userboot/devicename.c

199767f8919635c4928607450d9e0abb932109ceToomas Soome/*-
199767f8919635c4928607450d9e0abb932109ceToomas Soome * Copyright (c) 1998 Michael Smith <msmith@freebsd.org>
199767f8919635c4928607450d9e0abb932109ceToomas Soome * All rights reserved.
199767f8919635c4928607450d9e0abb932109ceToomas Soome *
199767f8919635c4928607450d9e0abb932109ceToomas Soome * Redistribution and use in source and binary forms, with or without
199767f8919635c4928607450d9e0abb932109ceToomas Soome * modification, are permitted provided that the following conditions
199767f8919635c4928607450d9e0abb932109ceToomas Soome * are met:
199767f8919635c4928607450d9e0abb932109ceToomas Soome * 1. Redistributions of source code must retain the above copyright
199767f8919635c4928607450d9e0abb932109ceToomas Soome *    notice, this list of conditions and the following disclaimer.
199767f8919635c4928607450d9e0abb932109ceToomas Soome * 2. Redistributions in binary form must reproduce the above copyright
199767f8919635c4928607450d9e0abb932109ceToomas Soome *    notice, this list of conditions and the following disclaimer in the
199767f8919635c4928607450d9e0abb932109ceToomas Soome *    documentation and/or other materials provided with the distribution.
199767f8919635c4928607450d9e0abb932109ceToomas Soome *
199767f8919635c4928607450d9e0abb932109ceToomas Soome * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
199767f8919635c4928607450d9e0abb932109ceToomas Soome * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
199767f8919635c4928607450d9e0abb932109ceToomas Soome * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
199767f8919635c4928607450d9e0abb932109ceToomas Soome * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
199767f8919635c4928607450d9e0abb932109ceToomas Soome * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
199767f8919635c4928607450d9e0abb932109ceToomas Soome * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
199767f8919635c4928607450d9e0abb932109ceToomas Soome * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
199767f8919635c4928607450d9e0abb932109ceToomas Soome * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
199767f8919635c4928607450d9e0abb932109ceToomas Soome * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
199767f8919635c4928607450d9e0abb932109ceToomas Soome * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
199767f8919635c4928607450d9e0abb932109ceToomas Soome * SUCH DAMAGE.
199767f8919635c4928607450d9e0abb932109ceToomas Soome */
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <sys/cdefs.h>
199767f8919635c4928607450d9e0abb932109ceToomas Soome__FBSDID("$FreeBSD$");
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <stand.h>
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <string.h>
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include "bootstrap.h"
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include "disk.h"
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include "libuserboot.h"
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome#if defined(USERBOOT_ZFS_SUPPORT)
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include "../zfs/libzfs.h"
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int  userboot_parsedev(struct disk_devdesc **dev, const char *devspec, const char **path);
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome/*
199767f8919635c4928607450d9e0abb932109ceToomas Soome * Point (dev) at an allocated device specifier for the device matching the
199767f8919635c4928607450d9e0abb932109ceToomas Soome * path in (devspec). If it contains an explicit device specification,
199767f8919635c4928607450d9e0abb932109ceToomas Soome * use that.  If not, use the default device.
199767f8919635c4928607450d9e0abb932109ceToomas Soome */
199767f8919635c4928607450d9e0abb932109ceToomas Soomeint
199767f8919635c4928607450d9e0abb932109ceToomas Soomeuserboot_getdev(void **vdev, const char *devspec, const char **path)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome    struct disk_devdesc **dev = (struct disk_devdesc **)vdev;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    int             rv;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    /*
199767f8919635c4928607450d9e0abb932109ceToomas Soome     * If it looks like this is just a path and no
199767f8919635c4928607450d9e0abb932109ceToomas Soome     * device, go with the current device.
199767f8919635c4928607450d9e0abb932109ceToomas Soome     */
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if ((devspec == NULL) ||
199767f8919635c4928607450d9e0abb932109ceToomas Soome    (devspec[0] == '/') ||
199767f8919635c4928607450d9e0abb932109ceToomas Soome    (strchr(devspec, ':') == NULL)) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (((rv = userboot_parsedev(dev, getenv("currdev"), NULL)) == 0) &&
199767f8919635c4928607450d9e0abb932109ceToomas Soome        (path != NULL))
199767f8919635c4928607450d9e0abb932109ceToomas Soome        *path = devspec;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return(rv);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    }
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    /*
199767f8919635c4928607450d9e0abb932109ceToomas Soome     * Try to parse the device name off the beginning of the devspec
199767f8919635c4928607450d9e0abb932109ceToomas Soome     */
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return(userboot_parsedev(dev, devspec, path));
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome/*
199767f8919635c4928607450d9e0abb932109ceToomas Soome * Point (dev) at an allocated device specifier matching the string version
199767f8919635c4928607450d9e0abb932109ceToomas Soome * at the beginning of (devspec).  Return a pointer to the remaining
199767f8919635c4928607450d9e0abb932109ceToomas Soome * text in (path).
199767f8919635c4928607450d9e0abb932109ceToomas Soome *
199767f8919635c4928607450d9e0abb932109ceToomas Soome * In all cases, the beginning of (devspec) is compared to the names
199767f8919635c4928607450d9e0abb932109ceToomas Soome * of known devices in the device switch, and then any following text
199767f8919635c4928607450d9e0abb932109ceToomas Soome * is parsed according to the rules applied to the device type.
199767f8919635c4928607450d9e0abb932109ceToomas Soome *
199767f8919635c4928607450d9e0abb932109ceToomas Soome * For disk-type devices, the syntax is:
199767f8919635c4928607450d9e0abb932109ceToomas Soome *
199767f8919635c4928607450d9e0abb932109ceToomas Soome * disk<unit>[s<slice>][<partition>]:
199767f8919635c4928607450d9e0abb932109ceToomas Soome *
199767f8919635c4928607450d9e0abb932109ceToomas Soome */
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int
199767f8919635c4928607450d9e0abb932109ceToomas Soomeuserboot_parsedev(struct disk_devdesc **dev, const char *devspec, const char **path)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome    struct disk_devdesc *idev;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    struct devsw    *dv;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    int         i, unit, err;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    const char      *cp;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    const char      *np;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    /* minimum length check */
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (strlen(devspec) < 2)
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return(EINVAL);
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    /* look for a device that matches */
199767f8919635c4928607450d9e0abb932109ceToomas Soome    for (i = 0, dv = NULL; devsw[i] != NULL; i++) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (!strncmp(devspec, devsw[i]->dv_name, strlen(devsw[i]->dv_name))) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome        dv = devsw[i];
199767f8919635c4928607450d9e0abb932109ceToomas Soome        break;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    }
199767f8919635c4928607450d9e0abb932109ceToomas Soome    }
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (dv == NULL)
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return(ENOENT);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    idev = malloc(sizeof(struct disk_devdesc));
199767f8919635c4928607450d9e0abb932109ceToomas Soome    err = 0;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    np = (devspec + strlen(dv->dv_name));
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    switch(dv->dv_type) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome    case DEVT_NONE:         /* XXX what to do here?  Do we care? */
199767f8919635c4928607450d9e0abb932109ceToomas Soome    break;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    case DEVT_DISK:
199767f8919635c4928607450d9e0abb932109ceToomas Soome    err = disk_parsedev(idev, np, path);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (err != 0)
199767f8919635c4928607450d9e0abb932109ceToomas Soome        goto fail;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    break;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    case DEVT_CD:
199767f8919635c4928607450d9e0abb932109ceToomas Soome    case DEVT_NET:
199767f8919635c4928607450d9e0abb932109ceToomas Soome    unit = 0;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (*np && (*np != ':')) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome        unit = strtol(np, (char **)&cp, 0); /* get unit number if present */
199767f8919635c4928607450d9e0abb932109ceToomas Soome        if (cp == np) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome        err = EUNIT;
199767f8919635c4928607450d9e0abb932109ceToomas Soome        goto fail;
199767f8919635c4928607450d9e0abb932109ceToomas Soome        }
199767f8919635c4928607450d9e0abb932109ceToomas Soome    } else {
199767f8919635c4928607450d9e0abb932109ceToomas Soome        cp = np;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    }
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (*cp && (*cp != ':')) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome        err = EINVAL;
199767f8919635c4928607450d9e0abb932109ceToomas Soome        goto fail;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    }
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    idev->d_unit = unit;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (path != NULL)
199767f8919635c4928607450d9e0abb932109ceToomas Soome        *path = (*cp == 0) ? cp : cp + 1;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    break;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    case DEVT_ZFS:
199767f8919635c4928607450d9e0abb932109ceToomas Soome#if defined(USERBOOT_ZFS_SUPPORT)
199767f8919635c4928607450d9e0abb932109ceToomas Soome        err = zfs_parsedev((struct zfs_devdesc *)idev, np, path);
199767f8919635c4928607450d9e0abb932109ceToomas Soome        if (err != 0)
199767f8919635c4928607450d9e0abb932109ceToomas Soome            goto fail;
199767f8919635c4928607450d9e0abb932109ceToomas Soome        break;
199767f8919635c4928607450d9e0abb932109ceToomas Soome#else
199767f8919635c4928607450d9e0abb932109ceToomas Soome        /* FALLTHROUGH */
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    default:
199767f8919635c4928607450d9e0abb932109ceToomas Soome    err = EINVAL;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    goto fail;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    }
199767f8919635c4928607450d9e0abb932109ceToomas Soome    idev->d_dev = dv;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    idev->d_type = dv->dv_type;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (dev == NULL) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome    free(idev);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    } else {
199767f8919635c4928607450d9e0abb932109ceToomas Soome    *dev = idev;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    }
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return(0);
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome fail:
199767f8919635c4928607450d9e0abb932109ceToomas Soome    free(idev);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return(err);
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomechar *
199767f8919635c4928607450d9e0abb932109ceToomas Soomeuserboot_fmtdev(void *vdev)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome    struct disk_devdesc *dev = (struct disk_devdesc *)vdev;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    static char     buf[128];   /* XXX device length constant? */
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    switch(dev->d_type) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome    case DEVT_NONE:
199767f8919635c4928607450d9e0abb932109ceToomas Soome    strcpy(buf, "(no device)");
199767f8919635c4928607450d9e0abb932109ceToomas Soome    break;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    case DEVT_CD:
199767f8919635c4928607450d9e0abb932109ceToomas Soome    sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    break;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    case DEVT_DISK:
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return (disk_fmtdev(vdev));
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    case DEVT_NET:
199767f8919635c4928607450d9e0abb932109ceToomas Soome    sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    break;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    case DEVT_ZFS:
199767f8919635c4928607450d9e0abb932109ceToomas Soome#if defined(USERBOOT_ZFS_SUPPORT)
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return (zfs_fmtdev(vdev));
199767f8919635c4928607450d9e0abb932109ceToomas Soome#else
199767f8919635c4928607450d9e0abb932109ceToomas Soome    sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome    break;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    }
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return(buf);
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome/*
199767f8919635c4928607450d9e0abb932109ceToomas Soome * Set currdev to suit the value being supplied in (value)
199767f8919635c4928607450d9e0abb932109ceToomas Soome */
199767f8919635c4928607450d9e0abb932109ceToomas Soomeint
199767f8919635c4928607450d9e0abb932109ceToomas Soomeuserboot_setcurrdev(struct env_var *ev, int flags, const void *value)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome    struct disk_devdesc *ncurr;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    int         rv;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if ((rv = userboot_parsedev(&ncurr, value, NULL)) != 0)
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return(rv);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    free(ncurr);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return(0);
199767f8919635c4928607450d9e0abb932109ceToomas Soome}